45 lines
1.5 KiB
Cython
45 lines
1.5 KiB
Cython
|
|
||
|
|
# WARNING: Do not edit this .pyx file directly, it is generated from its .pyx.tp
|
||
|
|
|
||
|
|
cdef class WeightVector64(object):
|
||
|
|
cdef readonly double[::1] w
|
||
|
|
cdef readonly double[::1] aw
|
||
|
|
cdef double *w_data_ptr
|
||
|
|
cdef double *aw_data_ptr
|
||
|
|
cdef double wscale
|
||
|
|
cdef double average_a
|
||
|
|
cdef double average_b
|
||
|
|
cdef int n_features
|
||
|
|
cdef double sq_norm
|
||
|
|
|
||
|
|
cdef void add(self, double *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz, double c) nogil
|
||
|
|
cdef void add_average(self, double *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz, double c, double num_iter) nogil
|
||
|
|
cdef double dot(self, double *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz) nogil
|
||
|
|
cdef void scale(self, double c) nogil
|
||
|
|
cdef void reset_wscale(self) nogil
|
||
|
|
cdef double norm(self) nogil
|
||
|
|
|
||
|
|
cdef class WeightVector32(object):
|
||
|
|
cdef readonly float[::1] w
|
||
|
|
cdef readonly float[::1] aw
|
||
|
|
cdef float *w_data_ptr
|
||
|
|
cdef float *aw_data_ptr
|
||
|
|
cdef float wscale
|
||
|
|
cdef float average_a
|
||
|
|
cdef float average_b
|
||
|
|
cdef int n_features
|
||
|
|
cdef float sq_norm
|
||
|
|
|
||
|
|
cdef void add(self, float *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz, float c) nogil
|
||
|
|
cdef void add_average(self, float *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz, float c, float num_iter) nogil
|
||
|
|
cdef float dot(self, float *x_data_ptr, int *x_ind_ptr,
|
||
|
|
int xnnz) nogil
|
||
|
|
cdef void scale(self, float c) nogil
|
||
|
|
cdef void reset_wscale(self) nogil
|
||
|
|
cdef float norm(self) nogil
|