OpenCV  2.4.13.2
Open Source Computer Vision
CvForestTree Class Reference

#include <ml.hpp>

Inheritance diagram for CvForestTree:
CvDTree CvStatModel CvForestERTree

Public Member Functions

 CvForestTree ()
 
virtual ~CvForestTree ()
 
virtual bool train (CvDTreeTrainData *trainData, const CvMat *_subsample_idx, CvRTrees *forest)
 
virtual int get_var_count () const
 
virtual void read (CvFileStorage *fs, CvFileNode *node, CvRTrees *forest, CvDTreeTrainData *_data)
 
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 void read (CvFileStorage *fs, CvFileNode *node)
 
virtual void read (CvFileStorage *fs, CvFileNode *node, CvDTreeTrainData *data)
 
- Public Member Functions inherited from CvDTree
 CvDTree ()
 
virtual ~CvDTree ()
 
virtual bool train (CvMLData *trainData, CvDTreeParams params=CvDTreeParams())
 
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 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 CvDTreeNodepredict (const cv::Mat &sample, const cv::Mat &missingDataMask=cv::Mat(), bool preprocessedInput=false) const
 
virtual cv::Mat getVarImportance ()
 
virtual const CvMatget_var_importance ()
 
virtual 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 ()
 
- Public Member Functions inherited from CvStatModel
 CvStatModel ()
 
virtual ~CvStatModel ()
 
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 CvDTreeSplitfind_best_split (CvDTreeNode *n)
 
- Protected Member Functions inherited from CvDTree
virtual bool do_train (const CvMat *_subsample_idx)
 
virtual void try_split_node (CvDTreeNode *n)
 
virtual void split_node_data (CvDTreeNode *n)
 
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 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 double calc_node_dir (CvDTreeNode *node)
 
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
 
- Protected Attributes inherited from CvDTree
CvDTreeNoderoot
 
CvMatvar_importance
 
CvDTreeTrainDatadata
 
- Protected Attributes inherited from CvStatModel
const char * default_model_name
 

Friends

struct cv::ForestTreeBestSplitFinder
 

Additional Inherited Members

- Public Attributes inherited from CvDTree
int pruned_tree_idx
 

Constructor & Destructor Documentation

§ CvForestTree()

CvForestTree::CvForestTree ( )

§ ~CvForestTree()

virtual CvForestTree::~CvForestTree ( )
virtual

Member Function Documentation

§ find_best_split()

virtual CvDTreeSplit* CvForestTree::find_best_split ( CvDTreeNode n)
protectedvirtual

Reimplemented from CvDTree.

§ get_var_count()

virtual int CvForestTree::get_var_count ( ) const
inlinevirtual

§ read() [1/3]

virtual void CvForestTree::read ( CvFileStorage fs,
CvFileNode node,
CvRTrees forest,
CvDTreeTrainData _data 
)
virtual

§ read() [2/3]

virtual void CvForestTree::read ( CvFileStorage fs,
CvFileNode node 
)
virtual

Reimplemented from CvDTree.

§ read() [3/3]

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

Reimplemented from CvDTree.

§ train() [1/3]

virtual bool CvForestTree::train ( CvDTreeTrainData trainData,
const CvMat _subsample_idx,
CvRTrees forest 
)
virtual

§ train() [2/3]

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

Reimplemented from CvDTree.

§ train() [3/3]

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

Reimplemented from CvDTree.

Friends And Related Function Documentation

§ cv::ForestTreeBestSplitFinder

friend struct cv::ForestTreeBestSplitFinder
friend

Member Data Documentation

§ forest

CvRTrees* CvForestTree::forest
protected

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