mirror of
https://github.com/explosion/spaCy.git
synced 2025-02-06 14:40:34 +03:00
Fix parser model
This commit is contained in:
parent
396dd60b3a
commit
2bcb5881d7
|
@ -261,7 +261,7 @@ class ParserStepModel(Model):
|
||||||
|
|
||||||
def get_token_ids(self, states):
|
def get_token_ids(self, states):
|
||||||
cdef StateClass state
|
cdef StateClass state
|
||||||
states = [state for state in states() if not state.is_final()]
|
states = [state for state in states if not state.is_final()]
|
||||||
cdef np.ndarray ids = numpy.zeros((len(states), self.state2vec.nF),
|
cdef np.ndarray ids = numpy.zeros((len(states), self.state2vec.nF),
|
||||||
dtype='i', order='C')
|
dtype='i', order='C')
|
||||||
ids.fill(-1)
|
ids.fill(-1)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user