explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Try to be aware of listeners in begin_training

Browse Source
This commit is contained in:
Matthew Honnibal 2020-08-28 16:16:03 +02:00
parent 1b8d2ed14f
commit 047f3ba10c
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/pipeline/tagger.pyx
View File

@ -293,9 +293,18 @@ class Tagger(Pipe):
] ]
for y in label_sample: for y in label_sample:
y[:, 0] = 1.0 y[:, 0] = 1.0
self.model.initialize(X=doc_sample, Y=label_sample)
else: else:
self.model.initialize() label_sample = None
if pipeline is not None:
for name, component in pipeline:
if component is self:
break
if hasattr(component, "pipe"):
doc_sample = list(component.pipe(doc_sample, batch_size=8))
else:
doc_sample = [component(doc) for doc in doc_sample]
self.model.initialize(X=doc_sample, Y=label_sample)
if sgd is None: if sgd is None:
sgd = self.create_optimizer() sgd = self.create_optimizer()
return sgd return sgd