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

spancat type fixes

Browse Source
This commit is contained in:
Daniël de Kok 2023-06-12 16:43:05 +02:00
parent d82e167aea
commit 4990cfefb4
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/pipeline/spancat.py
View File

@ -537,10 +537,11 @@ class SpanCategorizer(TrainablePipe):
DOCS: https://spacy.io/api/spancategorizer#set_annotations DOCS: https://spacy.io/api/spancategorizer#set_annotations
""" """
indices = activations["indices"] indices = activations["indices"]
scores = activations["scores"] assert isinstance(indices, Ragged)
scores = cast(Floats2d, activations["scores"])
offset = 0 offset = 0
for i, doc in enumerate(docs): for i, doc in enumerate(docs):
indices_i = indices[i].dataXd indices_i = cast(Ints2d, indices[i].dataXd)
if self.save_activations: if self.save_activations:
doc.activations[self.name] = {} doc.activations[self.name] = {}
doc.activations[self.name]["indices"] = indices_i doc.activations[self.name]["indices"] = indices_i