CvRTrees Class Reference

#include <ml.h>

Inherits CvStatModel.

Inherited by CvERTrees.

List of all members.

Public Member Functions

 CvRTrees ()
virtual ~CvRTrees ()
virtual bool train (const CvMat *_train_data, int _tflag, const CvMat *_responses, const CvMat *_var_idx=0, const CvMat *_sample_idx=0, const CvMat *_var_type=0, const CvMat *_missing_mask=0, CvRTParams params=CvRTParams())
virtual bool train (CvMLData *data, CvRTParams params=CvRTParams())
virtual float predict (const CvMat *sample, const CvMat *missing=0) const
virtual float predict_prob (const CvMat *sample, const CvMat *missing=0) const
virtual bool train (const cv::Mat &_train_data, int _tflag, const cv::Mat &_responses, const cv::Mat &_var_idx=cv::Mat(), const cv::Mat &_sample_idx=cv::Mat(), const cv::Mat &_var_type=cv::Mat(), const cv::Mat &_missing_mask=cv::Mat(), CvRTParams params=CvRTParams())
virtual float predict (const cv::Mat &sample, const cv::Mat &missing=cv::Mat()) const
virtual float predict_prob (const cv::Mat &sample, const cv::Mat &missing=cv::Mat()) const
virtual void clear ()
virtual const CvMatget_var_importance ()
virtual float get_proximity (const CvMat *sample1, const CvMat *sample2, const CvMat *missing1=0, const CvMat *missing2=0) const
virtual float calc_error (CvMLData *_data, int type, std::vector< float > *resp=0)
virtual float get_train_error ()
virtual void read (CvFileStorage *fs, CvFileNode *node)
virtual void write (CvFileStorage *fs, const char *name) const
CvMatget_active_var_mask ()
CvRNGget_rng ()
int get_tree_count () const
CvForestTreeget_tree (int i) const
virtual void save (const char *filename, const char *name=0) const
virtual void load (const char *filename, const char *name=0)

Protected Member Functions

virtual bool grow_forest (const CvTermCriteria term_crit)

Protected Attributes

CvForestTree ** trees
CvDTreeTrainDatadata
int ntrees
int nclasses
double oob_error
CvMatvar_importance
int nsamples
CvRNG rng
CvMatactive_var_mask
const char * default_model_name

Constructor & Destructor Documentation

CvRTrees::CvRTrees (  )
virtual CvRTrees::~CvRTrees (  ) [virtual]

Member Function Documentation

virtual bool CvRTrees::train ( const CvMat _train_data,
int  _tflag,
const CvMat _responses,
const CvMat _var_idx = 0,
const CvMat _sample_idx = 0,
const CvMat _var_type = 0,
const CvMat _missing_mask = 0,
CvRTParams  params = CvRTParams() 
) [virtual]

Reimplemented in CvERTrees.

virtual bool CvRTrees::train ( CvMLData data,
CvRTParams  params = CvRTParams() 
) [virtual]

Reimplemented in CvERTrees.

virtual float CvRTrees::predict ( const CvMat sample,
const CvMat missing = 0 
) const [virtual]
virtual float CvRTrees::predict_prob ( const CvMat sample,
const CvMat missing = 0 
) const [virtual]
virtual bool CvRTrees::train ( const cv::Mat _train_data,
int  _tflag,
const cv::Mat _responses,
const cv::Mat _var_idx = cv::Mat(),
const cv::Mat _sample_idx = cv::Mat(),
const cv::Mat _var_type = cv::Mat(),
const cv::Mat _missing_mask = cv::Mat(),
CvRTParams  params = CvRTParams() 
) [virtual]

Reimplemented in CvERTrees.

virtual float CvRTrees::predict ( const cv::Mat sample,
const cv::Mat missing = cv::Mat() 
) const [virtual]
virtual float CvRTrees::predict_prob ( const cv::Mat sample,
const cv::Mat missing = cv::Mat() 
) const [virtual]
virtual void CvRTrees::clear (  ) [virtual]

Reimplemented from CvStatModel.

virtual const CvMat* CvRTrees::get_var_importance (  ) [virtual]
virtual float CvRTrees::get_proximity ( const CvMat sample1,
const CvMat sample2,
const CvMat missing1 = 0,
const CvMat missing2 = 0 
) const [virtual]
virtual float CvRTrees::calc_error ( CvMLData _data,
int  type,
std::vector< float > *  resp = 0 
) [virtual]
virtual float CvRTrees::get_train_error (  ) [virtual]
virtual void CvRTrees::read ( CvFileStorage fs,
CvFileNode node 
) [virtual]

Reimplemented from CvStatModel.

virtual void CvRTrees::write ( CvFileStorage fs,
const char *  name 
) const [virtual]

Reimplemented from CvStatModel.

CvMat* CvRTrees::get_active_var_mask (  )
CvRNG* CvRTrees::get_rng (  )
int CvRTrees::get_tree_count (  ) const
CvForestTree* CvRTrees::get_tree ( int  i ) const
virtual bool CvRTrees::grow_forest ( const CvTermCriteria  term_crit ) [protected, virtual]

Reimplemented in CvERTrees.

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

int CvRTrees::ntrees [protected]
int CvRTrees::nclasses [protected]
double CvRTrees::oob_error [protected]
int CvRTrees::nsamples [protected]
CvRNG CvRTrees::rng [protected]
const char* CvStatModel::default_model_name [protected, inherited]

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