Don't set tensor in textcat

This commit is contained in:
Matthew Honnibal 2017-11-06 19:20:14 +01:00
parent 7d46793dd7
commit 8fea512ac8
1 changed files with 0 additions and 2 deletions

View File

@ -810,8 +810,6 @@ class TextCategorizer(Pipe):
def set_annotations(self, docs, scores, tensors=None):
for i, doc in enumerate(docs):
if tensors is not None:
doc.extend_tensor(tensors[i])
for j, label in enumerate(self.labels):
doc.cats[label] = float(scores[i, j])