explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 09:56:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

tensorizer return parameter fix

Browse Source
This commit is contained in:
uwol 2017-11-05 12:25:10 +01:00
parent 0a27afbf86
commit a2162b8908
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/pipeline.pyx
View File

@ -777,7 +777,8 @@ class TextCategorizer(Pipe):
def predict(self, docs): def predict(self, docs):
scores = self.model(docs) scores = self.model(docs)
scores = self.model.ops.asarray(scores) scores = self.model.ops.asarray(scores)
return scores tensors = [doc.tensor for doc in docs]
return scores, tensors
def set_annotations(self, docs, scores, tensors=None): def set_annotations(self, docs, scores, tensors=None):
for i, doc in enumerate(docs): for i, doc in enumerate(docs):