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

Fix parser

Browse Source
This commit is contained in:
Matthew Honnibal 2018-05-08 00:27:26 +02:00
parent 8a82367a9d
commit 99649d114d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/syntax/nn_parser.pyx
View File

@ -218,9 +218,10 @@ cdef class Parser:
if beam_width == 1: if beam_width == 1:
batch = self.moves.init_batch(docs) batch = self.moves.init_batch(docs)
weights = get_c_weights(model) weights = get_c_weights(model)
sizes = get_c_sizes(model, states.size())
for state in batch: for state in batch:
states.push_back(state.c) if not state.is_final():
states.push_back(state.c)
sizes = get_c_sizes(model, states.size())
with nogil: with nogil:
self._parseC(&states[0], self._parseC(&states[0],
weights, sizes) weights, sizes)