mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-24 17:06:29 +03:00
28 lines
623 B
Cython
28 lines
623 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 .moves cimport Move
|
|
|
|
|
|
cdef class NERParser:
|
|
cdef Pool mem
|
|
cdef Extractor extractor
|
|
cdef LinearModel model
|
|
cdef readonly list tag_names
|
|
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 int train(self, Tokens tokens, golds) except -1
|
|
cpdef int set_tags(self, Tokens tokens) except -1
|