explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-12 17:22:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

Default to beam_update_prob 1

Browse Source
This commit is contained in:
Matthew Honnibal 2018-05-10 15:38:02 +02:00
parent a61fd60681
commit 8e8724b55b
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/syntax/nn_parser.pyx
View File

@ -356,7 +356,7 @@ cdef class Parser:
losses.setdefault(self.name, 0.) losses.setdefault(self.name, 0.)
# 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', 0.5) beam_update_prob = self.cfg.get('beam_update_prob', 1.0)
if self.cfg.get('beam_width', 1) >= 2 and numpy.random.random() < beam_update_prob: if self.cfg.get('beam_width', 1) >= 2 and numpy.random.random() < beam_update_prob:
return self.update_beam(docs, golds, return self.update_beam(docs, golds,
self.cfg['beam_width'], self.cfg['beam_width'],
@ -391,7 +391,7 @@ cdef class Parser:
self.moves.preprocess_gold(gold) self.moves.preprocess_gold(gold)
model, finish_update = self.model.begin_update(docs, drop=drop) model, finish_update = self.model.begin_update(docs, drop=drop)
states_d_scores, backprops, beams = _beam_utils.update_beam( states_d_scores, backprops, beams = _beam_utils.update_beam(
self.moves, self.nr_feature, 500, states, golds, model.state2vec, self.moves, self.nr_feature, 10000, states, golds, model.state2vec,
model.vec2scores, width, drop=drop, losses=losses) model.vec2scores, width, drop=drop, losses=losses)
for i, d_scores in enumerate(states_d_scores): for i, d_scores in enumerate(states_d_scores):
losses[self.name] += (d_scores**2).sum() losses[self.name] += (d_scores**2).sum()