explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-22 14:30:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update tagger.begin_training

Browse Source
This commit is contained in:
Matthew Honnibal 2020-08-28 15:05:01 +02:00
parent c558ca4485
commit 237cfa2053
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/pipeline/tagger.pyx
View File

@ -287,7 +287,13 @@ class Tagger(Pipe):
self.add_label(tag) self.add_label(tag)
self.set_output(len(self.labels)) self.set_output(len(self.labels))
if self.labels: if self.labels:
self.model.initialize(X=doc_sample) label_sample = [
self.model.ops.alloc2f(len(doc), len(self.labels))
for doc in docs
]
for y in label_sample:
y[:, 0] = 1.0
self.model.initialize(X=doc_sample, Y=label_sample)
else: else:
self.model.initialize() self.model.initialize()
if sgd is None: if sgd is None: