2017-03-14 23:28:43 +03:00
|
|
|
from thinc.linear.features cimport ConjunctionExtracter
|
|
|
|
|
from thinc.typedefs cimport atom_t, weight_t
|
|
|
|
|
from thinc.structs cimport FeatureC
|
2017-03-24 02:23:32 +03:00
|
|
|
from libc.stdint cimport uint32_t, uint64_t
|
2017-03-14 23:28:43 +03:00
|
|
|
cimport numpy as np
|
|
|
|
|
from cymem.cymem cimport Pool
|
2017-03-24 02:23:32 +03:00
|
|
|
from libcpp.vector cimport vector
|
2017-03-14 23:28:43 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
cdef class LinearModel:
|
|
|
|
|
cdef ConjunctionExtracter extracter
|
|
|
|
|
cdef readonly int nr_class
|
|
|
|
|
cdef readonly uint32_t nr_weight
|
|
|
|
|
cdef public weight_t learn_rate
|
|
|
|
|
cdef Pool mem
|
|
|
|
|
cdef weight_t* W
|
|
|
|
|
cdef weight_t* d_W
|
2017-03-24 02:23:32 +03:00
|
|
|
cdef vector[uint64_t]* _indices
|
2017-03-14 23:28:43 +03:00
|
|
|
|
|
|
|
|
cdef void hinge_lossC(self, weight_t* d_scores,
|
|
|
|
|
const weight_t* scores, const weight_t* costs) nogil
|
|
|
|
|
|
|
|
|
|
cdef void log_lossC(self, weight_t* d_scores,
|
|
|
|
|
const weight_t* scores, const weight_t* costs) nogil
|
|
|
|
|
|
|
|
|
|
cdef void regression_lossC(self, weight_t* d_scores,
|
|
|
|
|
const weight_t* scores, const weight_t* costs) nogil
|
|
|
|
|
|
|
|
|
|
cdef void set_scoresC(self, weight_t* scores,
|
|
|
|
|
const FeatureC* features, int nr_feat) nogil
|
|
|
|
|
|
|
|
|
|
cdef void set_gradientC(self, const weight_t* d_scores, const FeatureC*
|
|
|
|
|
features, int nr_feat) nogil
|
