2012-04-13 04:27:02 +08:00
|
|
|
cimport numpy as np
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
cdef extern from "src/liblinear/linear.h":
|
|
|
|
|
cdef struct feature_node
|
|
|
|
|
cdef struct problem
|
|
|
|
|
cdef struct model
|
|
|
|
|
cdef struct parameter
|
2012-09-12 22:00:55 +08:00
|
|
|
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)
|
2013-04-14 08:38:43 +08:00
|
|
|
model *train(problem_const_ptr prob, parameter_const_ptr param) nogil
|
2012-04-13 04:27:02 +08:00
|
|
|
int get_nr_feature (model *model)
|
|
|
|
|
int get_nr_class (model *model)
|
2014-07-31 23:32:55 +08:00
|
|
|
void get_n_iter (model *model, int *n_iter)
|
2012-04-13 04:27:02 +08:00
|
|
|
void free_and_destroy_model (model **)
|
|
|
|
|
void destroy_param (parameter *)
|
|
|
|
|
|
|
|
|
|
cdef extern from "src/liblinear/liblinear_helper.c":
|
2012-09-12 22:00:55 +08:00
|
|
|
void copy_w(void *, model *, int)
|
2014-11-24 23:47:03 +08:00
|
|
|
parameter *set_parameter(int, double, double, int, char *, char *, int, int, double)
|
2015-10-20 02:46:15 +08:00
|
|
|
problem *set_problem (char *, char *, np.npy_intp *, double, double *)
|
2012-04-13 04:27:02 +08:00
|
|
|
problem *csr_set_problem (char *values, np.npy_intp *n_indices,
|
|
|
|
|
char *indices, np.npy_intp *n_indptr, char *indptr, char *Y,
|
2015-10-20 02:46:15 +08:00
|
|
|
np.npy_intp n_features, double bias, double *)
|
2012-04-13 04:27:02 +08:00
|
|
|
|
|
|
|
|
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)
|
