explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 19:39:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix beam parse. Not sure if working

Browse Source
This commit is contained in:
Matthew Honnibal 2017-08-12 18:21:54 -05:00
parent 24b45b45c6
commit c96d769836
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/syntax/_beam_utils.pyx
View File

@ -87,6 +87,7 @@ cdef class ParserBeam(object):
def _set_scores(self, Beam beam, scores):
for i in range(beam.size):
state = <StateClass>beam.at(i)
if not state.is_final():
for j in range(beam.nr_class):
beam.scores[i][j] = scores[i, j]
self.moves.set_valid(beam.is_valid[i], state.c)
@ -137,8 +138,8 @@ def update_beam(TransitionSystem moves, int nr_feature, int max_steps,
backprops.append((token_ids, bp_vectors, bp_scores))
p_scores = [scores[indices] for indices in p_indices]
g_scores = [scores[indices] for indices in g_indices]
p_scores = [numpy.ascontiguousarray(scores[indices], dtype='f') for indices in p_indices]
g_scores = [numpy.ascontiguousarray(scores[indices], dtype='f') for indices in g_indices]
pbeam.advance(p_scores)
gbeam.advance(g_scores, follow_gold=True)
@ -176,8 +177,8 @@ def get_states(pbeams, gbeams, beam_map):
beam_map[key] = len(states)
states.append(<StateClass>gbeam.at(i))
p_indices = numpy.asarray(p_indices, dtype='i')
g_indices = numpy.asarray(g_indices, dtype='i')
p_indices = [numpy.asarray(idx, dtype='i') for idx in p_indices]
g_indices = [numpy.asarray(idx, dtype='i') for idx in g_indices]
return states, p_indices, g_indices
@ -203,7 +204,9 @@ def get_gradient(nr_class, beam_maps, histories, losses):
key = tuple([eg_id])
for j, clas in enumerate(hist):
i = beam_maps[j][key]
grads[j][i, clas] = loss
# In step j, at state i action clas
# resulted in loss
grads[j][i, clas] += loss
key = key + tuple([clas])
return grads