Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix error when tagger was initialized with no labels

This commit is contained in:
Matthew Honnibal 2020-08-27 18:56:58 +02:00
parent e1e1760fd6
commit d3ffe4ca63
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/pipeline/tagger.pyx
View File

@ -286,7 +286,10 @@ class Tagger(Pipe):
for tag in sorted(tags): for tag in sorted(tags):
self.add_label(tag) self.add_label(tag)
self.set_output(len(self.labels)) self.set_output(len(self.labels))
if self.labels:
self.model.initialize(X=doc_sample) self.model.initialize(X=doc_sample)
else:
self.model.initialize()
if sgd is None: if sgd is None:
sgd = self.create_optimizer() sgd = self.create_optimizer()
return sgd return sgd