Use dep and ent multi-task objectives for parser'

This commit is contained in:
Matthew Honnibal 2017-09-26 08:13:52 -05:00
parent 9bfd585a11
commit ca28590ddd

View File

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