mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-21 19:39:13 +03:00
36 lines
990 B
Cython
36 lines
990 B
Cython
from libc.stdint cimport uint8_t
|
|
|
|
from cymem.cymem cimport Pool
|
|
|
|
from thinc.learner cimport LinearModel
|
|
from thinc.features cimport Extractor, Feature
|
|
from thinc.typedefs cimport atom_t, feat_t, weight_t, class_t
|
|
|
|
from preshed.maps cimport PreshMapArray
|
|
|
|
from .typedefs cimport hash_t, id_t
|
|
from .tokens cimport Tokens
|
|
|
|
|
|
cdef int arg_max(const weight_t* scores, const int n_classes) nogil
|
|
|
|
|
|
cdef class Model:
|
|
cdef int n_classes
|
|
|
|
cdef int regularize(self, Feature* feats, int n, int a=*) except -1
|
|
|
|
cdef int update(self, atom_t* context, class_t guess, class_t gold, int cost) except -1
|
|
|
|
cdef object model_loc
|
|
cdef Extractor _extractor
|
|
cdef LinearModel _model
|
|
|
|
cdef inline const weight_t* score(self, atom_t* context, bint regularize) except NULL:
|
|
cdef int n_feats
|
|
feats = self._extractor.get_feats(context, &n_feats)
|
|
if regularize:
|
|
self.regularize(feats, n_feats, 3)
|
|
return self._model.get_scores(feats, n_feats)
|
|
|