mirror of https://github.com/explosion/spaCy.git
30 lines
691 B
Cython
30 lines
691 B
Cython
|
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 *
|
||
|
|
|
||
|
|
from .structs cimport Move
|
||
|
|
from .annot cimport NERAnnotation
|
||
|
|
|
||
|
|
|
||
|
|
cdef class NERParser:
|
||
|
|
cdef Pool mem
|
||
|
|
cdef Extractor extractor
|
||
|
|
cdef LinearModel model
|
||
|
|
cdef readonly list tag_names
|
||
|
|
cdef readonly list entity_types
|
||
|
|
cdef readonly int n_classes
|
||
|
|
|
||
|
|
cdef Move* _moves
|
||
|
|
cdef atom_t* _context
|
||
|
|
cdef feat_t* _feats
|
||
|
|
cdef weight_t* _values
|
||
|
|
cdef weight_t* _scores
|
||
|
|
|
||
|
|
|
||
|
|
cpdef list train(self, Tokens tokens, NERAnnotation annot)
|
||
|
|
cpdef list set_tags(self, Tokens tokens)
|