Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix beam history features

This commit is contained in:
Matthew Honnibal 2017-10-05 22:21:50 -05:00
parent 3db0a32fd6
commit 555d8c8bff
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/syntax/nn_parser.pyx
View File

@ -508,9 +508,9 @@ cdef class Parser:
if self.cfg.get('hist_size', 0): if self.cfg.get('hist_size', 0):
hists = numpy.asarray([st.history[:self.cfg['hist_size']] hists = numpy.asarray([st.history[:self.cfg['hist_size']]
for st in states], dtype='i') for st in states], dtype='i')
scores = vec2scores(vectors, drop=drop) scores = vec2scores((vectors, hists))
else: else:
scores = vec2scores(vectors, drop=drop) scores = vec2scores(vectors)
j = 0 j = 0
c_scores = <float*>scores.data c_scores = <float*>scores.data
for i in range(beam.size): for i in range(beam.size):
@ -723,7 +723,7 @@ cdef class Parser:
lower, stream, drop=0.0) lower, stream, drop=0.0)
return (tokvecs, bp_tokvecs), state2vec, upper return (tokvecs, bp_tokvecs), state2vec, upper
nr_feature = 8 nr_feature = 2
def get_token_ids(self, states): def get_token_ids(self, states):
cdef StateClass state cdef StateClass state