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

Fix support for parser multi-task objectives

Browse Source
This commit is contained in:
Matthew Honnibal 2018-11-30 19:53:31 +01:00
parent 323fc26880
commit 0369db75c1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/syntax/nn_parser.pyx
View File

@ -373,6 +373,8 @@ cdef class Parser:
if losses is None: if losses is None:
losses = {} losses = {}
losses.setdefault(self.name, 0.) losses.setdefault(self.name, 0.)
for multitask in self._multitasks:
multitask.update(docs, golds, drop=drop, sgd=sgd)
# The probability we use beam update, instead of falling back to # The probability we use beam update, instead of falling back to
# a greedy update # a greedy update
beam_update_prob = self.cfg.get('beam_update_prob', 1.0) beam_update_prob = self.cfg.get('beam_update_prob', 1.0)