CvForestTree Class Reference

#include <ml.hpp>

Inheritance diagram for CvForestTree:

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 CvDTreeNode *	predict (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 CvDTreeNode *	predict (const cv::Mat &sample, const cv::Mat &missingDataMask=cv::Mat(), bool preprocessedInput=false) const
virtual cv::Mat	getVarImportance ()
virtual const CvMat *	get_var_importance ()
virtual void	clear ()
virtual void	write (CvFileStorage *fs, const char *name) const
virtual void	write (CvFileStorage *fs) const
const CvDTreeNode *	get_root () const
int	get_pruned_tree_idx () const
CvDTreeTrainData *	get_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 CvDTreeSplit *	find_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 CvDTreeSplit *	find_split_ord_class (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplit *	find_split_cat_class (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplit *	find_split_ord_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplit *	find_split_cat_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0)
virtual CvDTreeSplit *	find_surrogate_split_ord (CvDTreeNode *n, int vi, uchar *ext_buf=0)
virtual CvDTreeSplit *	find_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 CvDTreeNode *	read_node (CvFileStorage *fs, CvFileNode *node, CvDTreeNode *parent)
virtual CvDTreeSplit *	read_split (CvFileStorage *fs, CvFileNode *node)
virtual void	write_tree_nodes (CvFileStorage *fs) const
virtual void	read_tree_nodes (CvFileStorage *fs, CvFileNode *node)

Protected Attributes
CvRTrees *	forest
Protected Attributes inherited from CvDTree
CvDTreeNode *	root
CvMat *	var_importance
CvDTreeTrainData *	data
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:

• ml/ml.hpp