From 3cb8f06881f5991e4de2be4520e201612debb911 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 6 Aug 2017 14:15:14 +0200 Subject: [PATCH] Fix NeuralLabeller --- spacy/pipeline.pyx | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index 848653c5c..b87f73c27 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -296,7 +296,6 @@ class NeuralTagger(BaseThincComponent): if self.model.nI is None: self.model.nI = tokvecs[0].shape[1] - tag_scores, bp_tag_scores = self.model.begin_update(docs_tokvecs, drop=drop) loss, d_tag_scores = self.get_loss(docs, golds, tag_scores) @@ -432,7 +431,7 @@ class NeuralLabeller(NeuralTagger): @property def labels(self): - return self.cfg.get('labels', {}) + return self.cfg.setdefault('labels', {}) @labels.setter def labels(self, value):