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

Use dep and ent multi-task objectives for parser'

Browse Source
This commit is contained in:
Matthew Honnibal 2017-09-26 08:13:52 -05:00
parent 9bfd585a11
commit ca28590ddd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline.pyx
View File

@ -747,7 +747,7 @@ cdef class NeuralDependencyParser(NeuralParser):
TransitionSystem = ArcEager TransitionSystem = ArcEager
def init_multitask_objectives(self, gold_tuples, pipeline, **cfg): def init_multitask_objectives(self, gold_tuples, pipeline, **cfg):
for target in ['dep']: for target in ['dep', 'ent']:
labeller = NeuralLabeller(self.vocab, target=target) labeller = NeuralLabeller(self.vocab, target=target)
tok2vec = self.model[0] tok2vec = self.model[0]
labeller.begin_training(gold_tuples, pipeline=pipeline, tok2vec=tok2vec) labeller.begin_training(gold_tuples, pipeline=pipeline, tok2vec=tok2vec)