From 8fea512ac8b35fb5d50635140f13831442ae7d40 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 6 Nov 2017 19:20:14 +0100 Subject: [PATCH] Don't set tensor in textcat --- spacy/pipeline.pyx | 2 -- 1 file changed, 2 deletions(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index 4f56bde67..d8e58d3d4 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -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])