2014-12-16 00:06:04 +03:00
|
|
|
from cymem.cymem cimport Pool
|
|
|
|
|
|
|
|
|
|
from thinc.typedefs cimport weight_t
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from ._state cimport State
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
cdef struct Transition:
|
|
|
|
|
int move
|
|
|
|
|
int label
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
cdef class TransitionSystem:
|
|
|
|
|
cdef Pool mem
|
|
|
|
|
cdef readonly int n_moves
|
2014-12-16 14:44:43 +03:00
|
|
|
cdef dict label_ids
|
2014-12-16 00:06:04 +03:00
|
|
|
|
|
|
|
|
cdef const Transition* _moves
|
|
|
|
|
|
|
|
|
|
cdef int best_valid(self, const weight_t* scores, const State* s) except -1
|
|
|
|
|
cdef int best_gold(self, const weight_t* scores, const State* s,
|
2014-12-17 13:09:29 +03:00
|
|
|
int* gold_heads, int* gold_labels) except -1
|
2014-12-16 00:06:04 +03:00
|
|
|
cdef int transition(self, State *s, const int clas) except -1
|
