CvSVM Class Reference

#include <ml.h>

Inherits CvStatModel.

List of all members.

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
CvMatclass_labels
int var_all
float ** sv
int sv_total
CvMatvar_idx
CvMatclass_weights
CvSVMDecisionFuncdecision_func
CvMemStoragestorage
CvSVMSolversolver
CvSVMKernelkernel
const char * default_model_name

Member Enumeration Documentation

anonymous enum
Enumerator:
C_SVC 
NU_SVC 
ONE_CLASS 
EPS_SVR 
NU_SVR 
anonymous enum
Enumerator:
LINEAR 
POLY 
RBF 
SIGMOID 
anonymous enum
Enumerator:
C 
GAMMA 
P 
NU 
COEF 
DEGREE 

Constructor & Destructor Documentation

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() 
)

Member Function Documentation

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]

Member Data Documentation

CvMat* CvSVM::class_labels [protected]
int CvSVM::var_all [protected]
float** CvSVM::sv [protected]
int CvSVM::sv_total [protected]
CvMat* CvSVM::var_idx [protected]
CvSVMSolver* CvSVM::solver [protected]
CvSVMKernel* CvSVM::kernel [protected]
const char* CvStatModel::default_model_name [protected, inherited]

The documentation for this class was generated from the following file: