explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-22 18:12:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

remove duplicate declaration

Browse Source
This commit is contained in:
kadarakos 2023-02-20 10:36:54 +00:00
parent afc3a5a4af
commit e847487ebb
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/pipeline/spancat.py
View File

@ -712,9 +712,8 @@ class SpanCategorizer(TrainablePipe):
) -> SpanGroup:
"""Find the argmax label for each span."""
# Handle cases when there are zero suggestions
spans = SpanGroup(doc, name=self.key)
if scores.size == 0:
return spans
return SpanGroup(doc, name=self.key)
scores = self.model.ops.to_numpy(scores)
indices = self.model.ops.to_numpy(indices)
predicted = scores.argmax(axis=1)