2015-02-21 18:38:18 +03:00
|
|
|
from cymem.cymem cimport Pool
|
|
|
|
from thinc.typedefs cimport weight_t
|
|
|
|
|
|
|
|
from ..structs cimport TokenC
|
2015-05-24 22:35:02 +03:00
|
|
|
from ..gold cimport GoldParse
|
2015-06-02 21:01:06 +03:00
|
|
|
from ..gold cimport GoldParseC
|
2015-03-14 18:06:35 +03:00
|
|
|
from ..strings cimport StringStore
|
2015-02-21 18:38:18 +03:00
|
|
|
|
2015-06-10 00:23:28 +03:00
|
|
|
from .stateclass cimport StateClass
|
2016-02-01 04:58:14 +03:00
|
|
|
from ._state cimport StateC
|
2015-06-10 00:23:28 +03:00
|
|
|
|
2015-02-21 18:38:18 +03:00
|
|
|
|
|
|
|
cdef struct Transition:
|
|
|
|
int clas
|
|
|
|
int move
|
|
|
|
int label
|
|
|
|
|
|
|
|
weight_t score
|
|
|
|
|
2016-02-01 04:58:14 +03:00
|
|
|
bint (*is_valid)(const StateC* state, int label) nogil
|
2016-01-30 16:31:12 +03:00
|
|
|
weight_t (*get_cost)(StateClass state, const GoldParseC* gold, int label) nogil
|
2016-02-01 04:58:14 +03:00
|
|
|
int (*do)(StateC* state, int label) nogil
|
2015-02-21 18:38:18 +03:00
|
|
|
|
|
|
|
|
2016-01-30 16:31:12 +03:00
|
|
|
ctypedef weight_t (*get_cost_func_t)(StateClass state, const GoldParseC* gold, int label) nogil
|
|
|
|
ctypedef weight_t (*move_cost_func_t)(StateClass state, const GoldParseC* gold) nogil
|
|
|
|
ctypedef weight_t (*label_cost_func_t)(StateClass state, const GoldParseC* gold, int label) nogil
|
2015-02-21 18:38:18 +03:00
|
|
|
|
2016-02-01 04:58:14 +03:00
|
|
|
ctypedef int (*do_func_t)(StateC* state, int label) nogil
|
2015-02-21 18:38:18 +03:00
|
|
|
|
|
|
|
|
|
|
|
cdef class TransitionSystem:
|
|
|
|
cdef Pool mem
|
2015-03-14 18:06:35 +03:00
|
|
|
cdef StringStore strings
|
2016-01-19 21:09:33 +03:00
|
|
|
cdef Transition* c
|
2015-02-23 22:04:53 +03:00
|
|
|
cdef readonly int n_moves
|
2016-01-19 21:09:33 +03:00
|
|
|
cdef int _size
|
2015-07-15 00:47:03 +03:00
|
|
|
cdef public int root_label
|
2015-07-19 16:18:17 +03:00
|
|
|
cdef public freqs
|
2015-02-21 18:38:18 +03:00
|
|
|
|
2016-02-01 10:34:55 +03:00
|
|
|
cdef int initialize_state(self, StateC* state) nogil
|
|
|
|
cdef int finalize_state(self, StateC* state) nogil
|
2015-03-10 20:00:23 +03:00
|
|
|
|
2015-03-09 14:06:01 +03:00
|
|
|
cdef int preprocess_gold(self, GoldParse gold) except -1
|
2015-04-19 11:31:31 +03:00
|
|
|
|
2015-03-09 14:06:01 +03:00
|
|
|
cdef Transition lookup_transition(self, object name) except *
|
2015-04-19 11:31:31 +03:00
|
|
|
|
2015-02-21 18:38:18 +03:00
|
|
|
cdef Transition init_transition(self, int clas, int move, int label) except *
|
|
|
|
|
2016-02-01 05:00:15 +03:00
|
|
|
cdef int set_valid(self, int* output, const StateC* st) nogil
|
2015-08-09 00:35:59 +03:00
|
|
|
|
2016-01-30 16:31:12 +03:00
|
|
|
cdef int set_costs(self, int* is_valid, weight_t* costs,
|
2015-06-26 07:25:36 +03:00
|
|
|
StateClass state, GoldParse gold) except -1
|