2017-05-13 22:55:01 +00:00
|
|
|
from thinc.typedefs cimport atom_t
|
|
|
|
|
|
|
|
from .stateclass cimport StateClass
|
|
|
|
from .arc_eager cimport TransitionSystem
|
|
|
|
from ..vocab cimport Vocab
|
|
|
|
from ..tokens.doc cimport Doc
|
|
|
|
from ..structs cimport TokenC
|
|
|
|
from ._state cimport StateC
|
2018-05-15 20:17:29 +00:00
|
|
|
from ._parser_model cimport WeightsC, ActivationsC, SizesC
|
2017-05-13 22:55:01 +00:00
|
|
|
|
|
|
|
|
|
|
|
cdef class Parser:
|
|
|
|
cdef readonly Vocab vocab
|
2017-05-15 19:46:08 +00:00
|
|
|
cdef public object model
|
2017-05-13 22:55:01 +00:00
|
|
|
cdef readonly TransitionSystem moves
|
|
|
|
cdef readonly object cfg
|
2017-09-26 10:42:52 +00:00
|
|
|
cdef public object _multitasks
|
2018-05-15 20:17:29 +00:00
|
|
|
|
|
|
|
cdef void _parseC(self, StateC** states,
|
|
|
|
WeightsC weights, SizesC sizes) nogil
|
|
|
|
|
|
|
|
cdef void c_transition_batch(self, StateC** states, const float* scores,
|
|
|
|
int nr_class, int batch_size) nogil
|
|
|
|
|