#include <ml.h>
Inherits CvStatModel.
Public Types | |
enum | { C_SVC = 100, NU_SVC = 101, ONE_CLASS = 102, EPS_SVR = 103, NU_SVR = 104 } |
enum | { LINEAR = 0, POLY = 1, RBF = 2, SIGMOID = 3 } |
enum | { C = 0, GAMMA = 1, P = 2, NU = 3, COEF = 4, DEGREE = 5 } |
Public Member Functions | |
CvSVM () | |
virtual | ~CvSVM () |
CvSVM (const CvMat *_train_data, const CvMat *_responses, const CvMat *_var_idx=0, const CvMat *_sample_idx=0, CvSVMParams _params=CvSVMParams()) | |
virtual bool | train (const CvMat *_train_data, const CvMat *_responses, const CvMat *_var_idx=0, const CvMat *_sample_idx=0, CvSVMParams _params=CvSVMParams()) |
virtual bool | train_auto (const CvMat *_train_data, const CvMat *_responses, const CvMat *_var_idx, const CvMat *_sample_idx, CvSVMParams _params, int k_fold=10, CvParamGrid C_grid=get_default_grid(CvSVM::C), CvParamGrid gamma_grid=get_default_grid(CvSVM::GAMMA), CvParamGrid p_grid=get_default_grid(CvSVM::P), CvParamGrid nu_grid=get_default_grid(CvSVM::NU), CvParamGrid coef_grid=get_default_grid(CvSVM::COEF), CvParamGrid degree_grid=get_default_grid(CvSVM::DEGREE)) |
virtual float | predict (const CvMat *_sample, bool returnDFVal=false) const |
CvSVM (const cv::Mat &_train_data, const cv::Mat &_responses, const cv::Mat &_var_idx=cv::Mat(), const cv::Mat &_sample_idx=cv::Mat(), CvSVMParams _params=CvSVMParams()) | |
virtual bool | train (const cv::Mat &_train_data, const cv::Mat &_responses, const cv::Mat &_var_idx=cv::Mat(), const cv::Mat &_sample_idx=cv::Mat(), CvSVMParams _params=CvSVMParams()) |
virtual bool | train_auto (const cv::Mat &_train_data, const cv::Mat &_responses, const cv::Mat &_var_idx, const cv::Mat &_sample_idx, CvSVMParams _params, int k_fold=10, CvParamGrid C_grid=get_default_grid(CvSVM::C), CvParamGrid gamma_grid=get_default_grid(CvSVM::GAMMA), CvParamGrid p_grid=get_default_grid(CvSVM::P), CvParamGrid nu_grid=get_default_grid(CvSVM::NU), CvParamGrid coef_grid=get_default_grid(CvSVM::COEF), CvParamGrid degree_grid=get_default_grid(CvSVM::DEGREE)) |
virtual float | predict (const cv::Mat &_sample, bool returnDFVal=false) const |
virtual int | get_support_vector_count () const |
virtual const float * | get_support_vector (int i) const |
virtual CvSVMParams | get_params () const |
virtual void | clear () |
virtual void | write (CvFileStorage *storage, const char *name) const |
virtual void | read (CvFileStorage *storage, CvFileNode *node) |
int | get_var_count () const |
virtual void | save (const char *filename, const char *name=0) const |
virtual void | load (const char *filename, const char *name=0) |
Static Public Member Functions | |
static CvParamGrid | get_default_grid (int param_id) |
Protected Member Functions | |
virtual bool | set_params (const CvSVMParams &_params) |
virtual bool | train1 (int sample_count, int var_count, const float **samples, const void *_responses, double Cp, double Cn, CvMemStorage *_storage, double *alpha, double &rho) |
virtual bool | do_train (int svm_type, int sample_count, int var_count, const float **samples, const CvMat *_responses, CvMemStorage *_storage, double *alpha) |
virtual void | create_kernel () |
virtual void | create_solver () |
virtual float | predict (const float *row_sample, int row_len, bool returnDFVal=false) const |
virtual void | write_params (CvFileStorage *fs) const |
virtual void | read_params (CvFileStorage *fs, CvFileNode *node) |
Protected Attributes | |
CvSVMParams | params |
CvMat * | class_labels |
int | var_all |
float ** | sv |
int | sv_total |
CvMat * | var_idx |
CvMat * | class_weights |
CvSVMDecisionFunc * | decision_func |
CvMemStorage * | storage |
CvSVMSolver * | solver |
CvSVMKernel * | kernel |
const char * | default_model_name |
CvSVM::CvSVM | ( | ) |
virtual CvSVM::~CvSVM | ( | ) | [virtual] |
CvSVM::CvSVM | ( | const CvMat * | _train_data, | |
const CvMat * | _responses, | |||
const CvMat * | _var_idx = 0 , |
|||
const CvMat * | _sample_idx = 0 , |
|||
CvSVMParams | _params = CvSVMParams() | |||
) |
CvSVM::CvSVM | ( | const cv::Mat & | _train_data, | |
const cv::Mat & | _responses, | |||
const cv::Mat & | _var_idx = cv::Mat() , |
|||
const cv::Mat & | _sample_idx = cv::Mat() , |
|||
CvSVMParams | _params = CvSVMParams() | |||
) |
virtual bool CvSVM::train | ( | const CvMat * | _train_data, | |
const CvMat * | _responses, | |||
const CvMat * | _var_idx = 0 , |
|||
const CvMat * | _sample_idx = 0 , |
|||
CvSVMParams | _params = CvSVMParams() | |||
) | [virtual] |
virtual bool CvSVM::train_auto | ( | const CvMat * | _train_data, | |
const CvMat * | _responses, | |||
const CvMat * | _var_idx, | |||
const CvMat * | _sample_idx, | |||
CvSVMParams | _params, | |||
int | k_fold = 10 , |
|||
CvParamGrid | C_grid = get_default_grid(CvSVM::C) , |
|||
CvParamGrid | gamma_grid = get_default_grid(CvSVM::GAMMA) , |
|||
CvParamGrid | p_grid = get_default_grid(CvSVM::P) , |
|||
CvParamGrid | nu_grid = get_default_grid(CvSVM::NU) , |
|||
CvParamGrid | coef_grid = get_default_grid(CvSVM::COEF) , |
|||
CvParamGrid | degree_grid = get_default_grid(CvSVM::DEGREE) | |||
) | [virtual] |
virtual float CvSVM::predict | ( | const CvMat * | _sample, | |
bool | returnDFVal = false | |||
) | const [virtual] |
virtual bool CvSVM::train | ( | const cv::Mat & | _train_data, | |
const cv::Mat & | _responses, | |||
const cv::Mat & | _var_idx = cv::Mat() , |
|||
const cv::Mat & | _sample_idx = cv::Mat() , |
|||
CvSVMParams | _params = CvSVMParams() | |||
) | [virtual] |
virtual bool CvSVM::train_auto | ( | const cv::Mat & | _train_data, | |
const cv::Mat & | _responses, | |||
const cv::Mat & | _var_idx, | |||
const cv::Mat & | _sample_idx, | |||
CvSVMParams | _params, | |||
int | k_fold = 10 , |
|||
CvParamGrid | C_grid = get_default_grid(CvSVM::C) , |
|||
CvParamGrid | gamma_grid = get_default_grid(CvSVM::GAMMA) , |
|||
CvParamGrid | p_grid = get_default_grid(CvSVM::P) , |
|||
CvParamGrid | nu_grid = get_default_grid(CvSVM::NU) , |
|||
CvParamGrid | coef_grid = get_default_grid(CvSVM::COEF) , |
|||
CvParamGrid | degree_grid = get_default_grid(CvSVM::DEGREE) | |||
) | [virtual] |
virtual float CvSVM::predict | ( | const cv::Mat & | _sample, | |
bool | returnDFVal = false | |||
) | const [virtual] |
virtual int CvSVM::get_support_vector_count | ( | ) | const [virtual] |
virtual const float* CvSVM::get_support_vector | ( | int | i | ) | const [virtual] |
virtual CvSVMParams CvSVM::get_params | ( | ) | const [virtual] |
virtual void CvSVM::clear | ( | ) | [virtual] |
Reimplemented from CvStatModel.
static CvParamGrid CvSVM::get_default_grid | ( | int | param_id | ) | [static] |
virtual void CvSVM::write | ( | CvFileStorage * | storage, | |
const char * | name | |||
) | const [virtual] |
Reimplemented from CvStatModel.
virtual void CvSVM::read | ( | CvFileStorage * | storage, | |
CvFileNode * | node | |||
) | [virtual] |
Reimplemented from CvStatModel.
int CvSVM::get_var_count | ( | ) | const |
virtual bool CvSVM::set_params | ( | const CvSVMParams & | _params | ) | [protected, virtual] |
virtual bool CvSVM::train1 | ( | int | sample_count, | |
int | var_count, | |||
const float ** | samples, | |||
const void * | _responses, | |||
double | Cp, | |||
double | Cn, | |||
CvMemStorage * | _storage, | |||
double * | alpha, | |||
double & | rho | |||
) | [protected, virtual] |
virtual bool CvSVM::do_train | ( | int | svm_type, | |
int | sample_count, | |||
int | var_count, | |||
const float ** | samples, | |||
const CvMat * | _responses, | |||
CvMemStorage * | _storage, | |||
double * | alpha | |||
) | [protected, virtual] |
virtual void CvSVM::create_kernel | ( | ) | [protected, virtual] |
virtual void CvSVM::create_solver | ( | ) | [protected, virtual] |
virtual float CvSVM::predict | ( | const float * | row_sample, | |
int | row_len, | |||
bool | returnDFVal = false | |||
) | const [protected, virtual] |
virtual void CvSVM::write_params | ( | CvFileStorage * | fs | ) | const [protected, virtual] |
virtual void CvSVM::read_params | ( | CvFileStorage * | fs, | |
CvFileNode * | node | |||
) | [protected, virtual] |
virtual void CvStatModel::save | ( | const char * | filename, | |
const char * | name = 0 | |||
) | const [virtual, inherited] |
virtual void CvStatModel::load | ( | const char * | filename, | |
const char * | name = 0 | |||
) | [virtual, inherited] |
CvSVMParams CvSVM::params [protected] |
CvMat* CvSVM::class_labels [protected] |
int CvSVM::var_all [protected] |
float** CvSVM::sv [protected] |
int CvSVM::sv_total [protected] |
CvMat* CvSVM::var_idx [protected] |
CvMat* CvSVM::class_weights [protected] |
CvSVMDecisionFunc* CvSVM::decision_func [protected] |
CvMemStorage* CvSVM::storage [protected] |
CvSVMSolver* CvSVM::solver [protected] |
CvSVMKernel* CvSVM::kernel [protected] |
const char* CvStatModel::default_model_name [protected, inherited] |