Public Member Functions | Public Attributes | Protected Member Functions | Protected Attributes | Friends
CvForestERTree Class Reference

#include <ml.hpp>

Inherits CvForestTree.

List of all members.

Public Member Functions

virtual bool train (CvDTreeTrainData *trainData, const CvMat *_subsample_idx, CvRTrees *forest)
virtual bool train (const CvMat *trainData, int tflag, const CvMat *responses, const CvMat *varIdx=0, const CvMat *sampleIdx=0, const CvMat *varType=0, const CvMat *missingDataMask=0, CvDTreeParams params=CvDTreeParams())
virtual bool train (CvDTreeTrainData *trainData, const CvMat *_subsample_idx)
virtual bool train (CvMLData *trainData, CvDTreeParams params=CvDTreeParams())
virtual CV_WRAP bool train (const cv::Mat &trainData, int tflag, const cv::Mat &responses, const cv::Mat &varIdx=cv::Mat(), const cv::Mat &sampleIdx=cv::Mat(), const cv::Mat &varType=cv::Mat(), const cv::Mat &missingDataMask=cv::Mat(), CvDTreeParams params=CvDTreeParams())
virtual int get_var_count () const
virtual void read (CvFileStorage *fs, CvFileNode *node, CvRTrees *forest, CvDTreeTrainData *_data)
virtual void read (CvFileStorage *fs, CvFileNode *node)
virtual void read (CvFileStorage *fs, CvFileNode *node, CvDTreeTrainData *data)
virtual float calc_error (CvMLData *trainData, int type, std::vector< float > *resp=0)
virtual CvDTreeNodepredict (const CvMat *sample, const CvMat *missingDataMask=0, bool preprocessedInput=false) const
virtual CV_WRAP CvDTreeNodepredict (const cv::Mat &sample, const cv::Mat &missingDataMask=cv::Mat(), bool preprocessedInput=false) const
virtual CV_WRAP cv::Mat getVarImportance ()
virtual const CvMatget_var_importance ()
virtual CV_WRAP void clear ()
virtual void write (CvFileStorage *fs, const char *name) const
virtual void write (CvFileStorage *fs) const
const CvDTreeNodeget_root () const
int get_pruned_tree_idx () const
CvDTreeTrainDataget_data ()
virtual CV_WRAP void save (const char *filename, const char *name=0) const
virtual CV_WRAP void load (const char *filename, const char *name=0)

Public Attributes

int pruned_tree_idx

Protected Member Functions

virtual double calc_node_dir (CvDTreeNode *node)
virtual CvDTreeSplitfind_split_ord_class (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplitfind_split_cat_class (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplitfind_split_ord_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplitfind_split_cat_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual void split_node_data (CvDTreeNode *n)
virtual CvDTreeSplitfind_best_split (CvDTreeNode *n)
virtual bool do_train (const CvMat *_subsample_idx)
virtual void try_split_node (CvDTreeNode *n)
virtual CvDTreeSplitfind_surrogate_split_ord (CvDTreeNode *n, int vi, uchar *ext_buf=0)
virtual CvDTreeSplitfind_surrogate_split_cat (CvDTreeNode *n, int vi, uchar *ext_buf=0)
virtual void complete_node_dir (CvDTreeNode *node)
virtual void cluster_categories (const int *vectors, int vector_count, int var_count, int *sums, int k, int *cluster_labels)
virtual void calc_node_value (CvDTreeNode *node)
virtual void prune_cv ()
virtual double update_tree_rnc (int T, int fold)
virtual int cut_tree (int T, int fold, double min_alpha)
virtual void free_prune_data (bool cut_tree)
virtual void free_tree ()
virtual void write_node (CvFileStorage *fs, CvDTreeNode *node) const
virtual void write_split (CvFileStorage *fs, CvDTreeSplit *split) const
virtual CvDTreeNoderead_node (CvFileStorage *fs, CvFileNode *node, CvDTreeNode *parent)
virtual CvDTreeSplitread_split (CvFileStorage *fs, CvFileNode *node)
virtual void write_tree_nodes (CvFileStorage *fs) const
virtual void read_tree_nodes (CvFileStorage *fs, CvFileNode *node)

Protected Attributes

CvRTreesforest
CvDTreeNoderoot
CvMatvar_importance
CvDTreeTrainDatadata
const char * default_model_name

Friends

struct cv::ForestTreeBestSplitFinder
struct cv::DTreeBestSplitFinder

Member Function Documentation

virtual double CvForestERTree::calc_node_dir ( CvDTreeNode node) [protected, virtual]

Reimplemented from CvDTree.

virtual CvDTreeSplit* CvForestERTree::find_split_ord_class ( CvDTreeNode n,
int  vi,
float  init_quality = 0,
CvDTreeSplit _split = 0,
uchar ext_buf = 0 
) [protected, virtual]

Reimplemented from CvDTree.

virtual CvDTreeSplit* CvForestERTree::find_split_cat_class ( CvDTreeNode n,
int  vi,
float  init_quality = 0,
CvDTreeSplit _split = 0,
uchar ext_buf = 0 
) [protected, virtual]

Reimplemented from CvDTree.

virtual CvDTreeSplit* CvForestERTree::find_split_ord_reg ( CvDTreeNode n,
int  vi,
float  init_quality = 0,
CvDTreeSplit _split = 0,
uchar ext_buf = 0 
) [protected, virtual]

Reimplemented from CvDTree.

virtual CvDTreeSplit* CvForestERTree::find_split_cat_reg ( CvDTreeNode n,
int  vi,
float  init_quality = 0,
CvDTreeSplit _split = 0,
uchar ext_buf = 0 
) [protected, virtual]

Reimplemented from CvDTree.

virtual void CvForestERTree::split_node_data ( CvDTreeNode n) [protected, virtual]

Reimplemented from CvDTree.

virtual bool CvForestTree::train ( CvDTreeTrainData trainData,
const CvMat _subsample_idx,
CvRTrees forest 
) [virtual, inherited]
virtual bool CvForestTree::train ( const CvMat trainData,
int  tflag,
const CvMat responses,
const CvMat varIdx = 0,
const CvMat sampleIdx = 0,
const CvMat varType = 0,
const CvMat missingDataMask = 0,
CvDTreeParams  params = CvDTreeParams() 
) [virtual, inherited]

Reimplemented from CvDTree.

virtual bool CvForestTree::train ( CvDTreeTrainData trainData,
const CvMat _subsample_idx 
) [virtual, inherited]

Reimplemented from CvDTree.

virtual bool CvDTree::train ( CvMLData trainData,
CvDTreeParams  params = CvDTreeParams() 
) [virtual, inherited]
virtual CV_WRAP bool CvDTree::train ( const cv::Mat trainData,
int  tflag,
const cv::Mat responses,
const cv::Mat varIdx = cv::Mat(),
const cv::Mat sampleIdx = cv::Mat(),
const cv::Mat varType = cv::Mat(),
const cv::Mat missingDataMask = cv::Mat(),
CvDTreeParams  params = CvDTreeParams() 
) [virtual, inherited]
virtual int CvForestTree::get_var_count ( ) const [virtual, inherited]
virtual void CvForestTree::read ( CvFileStorage fs,
CvFileNode node,
CvRTrees forest,
CvDTreeTrainData _data 
) [virtual, inherited]
virtual void CvForestTree::read ( CvFileStorage fs,
CvFileNode node 
) [virtual, inherited]

Reimplemented from CvDTree.

virtual void CvForestTree::read ( CvFileStorage fs,
CvFileNode node,
CvDTreeTrainData data 
) [virtual, inherited]

Reimplemented from CvDTree.

virtual CvDTreeSplit* CvForestTree::find_best_split ( CvDTreeNode n) [protected, virtual, inherited]

Reimplemented from CvDTree.

virtual float CvDTree::calc_error ( CvMLData trainData,
int  type,
std::vector< float > *  resp = 0 
) [virtual, inherited]
virtual CvDTreeNode* CvDTree::predict ( const CvMat sample,
const CvMat missingDataMask = 0,
bool  preprocessedInput = false 
) const [virtual, inherited]
virtual CV_WRAP CvDTreeNode* CvDTree::predict ( const cv::Mat sample,
const cv::Mat missingDataMask = cv::Mat(),
bool  preprocessedInput = false 
) const [virtual, inherited]
virtual CV_WRAP cv::Mat CvDTree::getVarImportance ( ) [virtual, inherited]
virtual const CvMat* CvDTree::get_var_importance ( ) [virtual, inherited]
virtual CV_WRAP void CvDTree::clear ( ) [virtual, inherited]

Reimplemented from CvStatModel.

Reimplemented in CvBoostTree.

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

Reimplemented from CvStatModel.

virtual void CvDTree::write ( CvFileStorage fs) const [virtual, inherited]
const CvDTreeNode* CvDTree::get_root ( ) const [inherited]
int CvDTree::get_pruned_tree_idx ( ) const [inherited]
virtual bool CvDTree::do_train ( const CvMat _subsample_idx) [protected, virtual, inherited]
virtual void CvDTree::try_split_node ( CvDTreeNode n) [protected, virtual, inherited]

Reimplemented in CvBoostTree.

virtual CvDTreeSplit* CvDTree::find_surrogate_split_ord ( CvDTreeNode n,
int  vi,
uchar ext_buf = 0 
) [protected, virtual, inherited]

Reimplemented in CvBoostTree.

virtual CvDTreeSplit* CvDTree::find_surrogate_split_cat ( CvDTreeNode n,
int  vi,
uchar ext_buf = 0 
) [protected, virtual, inherited]

Reimplemented in CvBoostTree.

virtual void CvDTree::complete_node_dir ( CvDTreeNode node) [protected, virtual, inherited]
virtual void CvDTree::cluster_categories ( const int *  vectors,
int  vector_count,
int  var_count,
int *  sums,
int  k,
int *  cluster_labels 
) [protected, virtual, inherited]
virtual void CvDTree::calc_node_value ( CvDTreeNode node) [protected, virtual, inherited]

Reimplemented in CvBoostTree.

virtual void CvDTree::prune_cv ( ) [protected, virtual, inherited]
virtual double CvDTree::update_tree_rnc ( int  T,
int  fold 
) [protected, virtual, inherited]
virtual int CvDTree::cut_tree ( int  T,
int  fold,
double  min_alpha 
) [protected, virtual, inherited]
virtual void CvDTree::free_prune_data ( bool  cut_tree) [protected, virtual, inherited]
virtual void CvDTree::free_tree ( ) [protected, virtual, inherited]
virtual void CvDTree::write_node ( CvFileStorage fs,
CvDTreeNode node 
) const [protected, virtual, inherited]
virtual void CvDTree::write_split ( CvFileStorage fs,
CvDTreeSplit split 
) const [protected, virtual, inherited]
virtual CvDTreeNode* CvDTree::read_node ( CvFileStorage fs,
CvFileNode node,
CvDTreeNode parent 
) [protected, virtual, inherited]
virtual CvDTreeSplit* CvDTree::read_split ( CvFileStorage fs,
CvFileNode node 
) [protected, virtual, inherited]
virtual void CvDTree::write_tree_nodes ( CvFileStorage fs) const [protected, virtual, inherited]
virtual void CvDTree::read_tree_nodes ( CvFileStorage fs,
CvFileNode node 
) [protected, virtual, inherited]
virtual CV_WRAP void CvStatModel::save ( const char *  filename,
const char *  name = 0 
) const [virtual, inherited]
virtual CV_WRAP void CvStatModel::load ( const char *  filename,
const char *  name = 0 
) [virtual, inherited]

Friends And Related Function Documentation

friend struct cv::ForestTreeBestSplitFinder [friend, inherited]
friend struct cv::DTreeBestSplitFinder [friend, inherited]

Member Data Documentation

CvRTrees* CvForestTree::forest [protected, inherited]
CvDTreeNode* CvDTree::root [protected, inherited]
CvMat* CvDTree::var_importance [protected, inherited]
CvDTreeTrainData* CvDTree::data [protected, inherited]
int CvDTree::pruned_tree_idx [inherited]
const char* CvStatModel::default_model_name [protected, inherited]

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