2014-11-10 05:28:56 +00:00
|
|
|
from cymem.cymem cimport Pool
|
|
|
|
|
from thinc.features cimport Extractor
|
|
|
|
|
from thinc.learner cimport LinearModel
|
|
|
|
|
from thinc.typedefs cimport *
|
|
|
|
|
|
|
|
|
|
from ..tokens cimport Tokens
|
|
|
|
|
from ..typedefs cimport *
|
|
|
|
|
|
2014-11-11 13:54:25 +00:00
|
|
|
from .bilou_moves cimport Move
|
2014-11-10 05:28:56 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
cdef class NERParser:
|
|
|
|
|
cdef Pool mem
|
|
|
|
|
cdef Extractor extractor
|
|
|
|
|
cdef LinearModel model
|
2014-11-11 06:17:54 +00:00
|
|
|
cdef readonly list tag_names
|
|
|
|
|
cdef readonly int n_classes
|
2014-11-10 05:28:56 +00:00
|
|
|
|
|
|
|
|
cdef Move* _moves
|
|
|
|
|
cdef atom_t* _context
|
|
|
|
|
cdef feat_t* _feats
|
|
|
|
|
cdef weight_t* _values
|
|
|
|
|
cdef weight_t* _scores
|
|
|
|
|
|
|
|
|
|
|
2014-11-11 06:17:54 +00:00
|
|
|
cpdef int train(self, Tokens tokens, golds) except -1
|
2014-11-10 05:28:56 +00:00
|
|
|
cpdef int set_tags(self, Tokens tokens) except -1
|
