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

Fix multi-task objective for parser

Browse Source
This commit is contained in:
Matthew Honnibal 2018-09-13 14:08:55 +02:00
parent 21321cd6cf
commit a95eea4c06
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline.pyx
View File

@ -1172,7 +1172,7 @@ cdef class DependencyParser(Parser):
def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg): def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg):
for labeller in self._multitasks: for labeller in self._multitasks:
tok2vec = self.model[0] tok2vec = self.model.tok2vec
labeller.begin_training(get_gold_tuples, pipeline=pipeline, labeller.begin_training(get_gold_tuples, pipeline=pipeline,
tok2vec=tok2vec, sgd=sgd) tok2vec=tok2vec, sgd=sgd)
@ -1193,7 +1193,7 @@ cdef class EntityRecognizer(Parser):
def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg): def init_multitask_objectives(self, get_gold_tuples, pipeline, sgd=None, **cfg):
for labeller in self._multitasks: for labeller in self._multitasks:
tok2vec = self.model[0] tok2vec = self.model.tok2vec
labeller.begin_training(get_gold_tuples, pipeline=pipeline, labeller.begin_training(get_gold_tuples, pipeline=pipeline,
tok2vec=tok2vec) tok2vec=tok2vec)