34 lines
1.3 KiB
Cython
34 lines
1.3 KiB
Cython
cimport numpy as np
|
|
|
|
|
|
cdef extern from "src/liblinear/linear.h":
|
|
cdef struct feature_node
|
|
cdef struct problem
|
|
cdef struct model
|
|
cdef struct parameter
|
|
ctypedef problem* problem_const_ptr "problem const *"
|
|
ctypedef parameter* parameter_const_ptr "parameter const *"
|
|
ctypedef char* char_const_ptr "char const *"
|
|
char_const_ptr check_parameter(problem_const_ptr prob, parameter_const_ptr param)
|
|
model *train(problem_const_ptr prob, parameter_const_ptr param) nogil
|
|
int get_nr_feature (model *model)
|
|
int get_nr_class (model *model)
|
|
void get_n_iter (model *model, int *n_iter)
|
|
void free_and_destroy_model (model **)
|
|
void destroy_param (parameter *)
|
|
|
|
cdef extern from "src/liblinear/liblinear_helper.c":
|
|
void copy_w(void *, model *, int)
|
|
parameter *set_parameter(int, double, double, int, char *, char *, int, int, double)
|
|
problem *set_problem (char *, char *, np.npy_intp *, double, char *)
|
|
problem *csr_set_problem (char *values, np.npy_intp *n_indices,
|
|
char *indices, np.npy_intp *n_indptr, char *indptr, char *Y,
|
|
np.npy_intp n_features, double bias, char *)
|
|
|
|
model *set_model(parameter *, char *, np.npy_intp *, char *, double)
|
|
|
|
double get_bias(model *)
|
|
void free_problem (problem *)
|
|
void free_parameter (parameter *)
|
|
void set_verbosity(int)
|