#include <ml.hpp>

Inheritance diagram for CvDTree:

Public Member Functions
	CvDTree ()

virtual	~CvDTree ()

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 (CvMLData *trainData, CvDTreeParams params=CvDTreeParams())

virtual float	calc_error (CvMLData trainData, int type, std::vector< float > resp=0)

virtual bool	train (CvDTreeTrainData trainData, const CvMat subsampleIdx)

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	read (CvFileStorage fs, CvFileNode node)

virtual void	write (CvFileStorage fs, const char name) const

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

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)

Public Attributes
int	pruned_tree_idx

Protected Member Functions
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_best_split (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
CvDTreeNode *	root

CvMat *	var_importance

CvDTreeTrainData *	data

Protected Attributes inherited from CvStatModel
const char *	default_model_name

Friends
struct	cv::DTreeBestSplitFinder

Constructor & Destructor Documentation

CvDTree::CvDTree ( )

virtual CvDTree::~CvDTree ( )

virtual

Member Function Documentation

virtual float CvDTree::calc_error	(	CvMLData *	trainData,
		int	type,
		std::vector< float > *	resp = `0`
	)

virtual

virtual double CvDTree::calc_node_dir ( CvDTreeNode * node )

protectedvirtual

Reimplemented in CvBoostTree, and CvForestERTree.

virtual void CvDTree::calc_node_value ( CvDTreeNode * node )

protectedvirtual

Reimplemented in CvBoostTree.

virtual void CvDTree::clear ( )

virtual

Reimplemented from CvStatModel.

Reimplemented in CvBoostTree.

virtual void CvDTree::cluster_categories	(	const int *	vectors,
		int	vector_count,
		int	var_count,
		int *	sums,
		int	k,
		int *	cluster_labels
	)

protectedvirtual

virtual void CvDTree::complete_node_dir ( CvDTreeNode * node )

protectedvirtual

virtual int CvDTree::cut_tree	(	int	T,
		int	fold,
		double	min_alpha
	)

protectedvirtual

virtual bool CvDTree::do_train ( const CvMat * _subsample_idx )

protectedvirtual

virtual CvDTreeSplit* CvDTree::find_best_split ( CvDTreeNode * n )

protectedvirtual

Reimplemented in CvForestTree.

virtual CvDTreeSplit* CvDTree::find_split_cat_class	(	CvDTreeNode *	n,
		int	vi,
		float	init_quality = `0`,
		CvDTreeSplit *	_split = `0`,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree, and CvForestERTree.

virtual CvDTreeSplit* CvDTree::find_split_cat_reg	(	CvDTreeNode *	n,
		int	vi,
		float	init_quality = `0`,
		CvDTreeSplit *	_split = `0`,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree, and CvForestERTree.

virtual CvDTreeSplit* CvDTree::find_split_ord_class	(	CvDTreeNode *	n,
		int	vi,
		float	init_quality = `0`,
		CvDTreeSplit *	_split = `0`,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree, and CvForestERTree.

virtual CvDTreeSplit* CvDTree::find_split_ord_reg	(	CvDTreeNode *	n,
		int	vi,
		float	init_quality = `0`,
		CvDTreeSplit *	_split = `0`,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree, and CvForestERTree.

virtual CvDTreeSplit* CvDTree::find_surrogate_split_cat	(	CvDTreeNode *	n,
		int	vi,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree.

virtual CvDTreeSplit* CvDTree::find_surrogate_split_ord	(	CvDTreeNode *	n,
		int	vi,
		uchar *	ext_buf = `0`
	)

protectedvirtual

Reimplemented in CvBoostTree.

virtual void CvDTree::free_prune_data ( bool cut_tree )

protectedvirtual

virtual void CvDTree::free_tree ( )

protectedvirtual

CvDTreeTrainData* CvDTree::get_data ( )

int CvDTree::get_pruned_tree_idx ( ) const

const CvDTreeNode* CvDTree::get_root ( ) const

virtual const CvMat* CvDTree::get_var_importance ( )

virtual

virtual cv::Mat CvDTree::getVarImportance ( )

virtual

virtual CvDTreeNode* CvDTree::predict	(	const CvMat *	sample,
		const CvMat *	missingDataMask = `0`,
		bool	preprocessedInput = `false`
	)		const

virtual

virtual CvDTreeNode* CvDTree::predict	(	const cv::Mat &	sample,
		const cv::Mat &	missingDataMask = `cv::Mat()`,
		bool	preprocessedInput = `false`
	)		const

virtual

virtual void CvDTree::prune_cv ( )

protectedvirtual

virtual void CvDTree::read	(	CvFileStorage *	fs,
		CvFileNode *	node
	)

virtual

Reimplemented from CvStatModel.

Reimplemented in CvBoostTree, and CvForestTree.

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

virtual

Reimplemented in CvBoostTree, and CvForestTree.

virtual CvDTreeNode* CvDTree::read_node	(	CvFileStorage *	fs,
		CvFileNode *	node,
		CvDTreeNode *	parent
	)

protectedvirtual

virtual CvDTreeSplit* CvDTree::read_split	(	CvFileStorage *	fs,
		CvFileNode *	node
	)

protectedvirtual

virtual void CvDTree::read_tree_nodes	(	CvFileStorage *	fs,
		CvFileNode *	node
	)

protectedvirtual

virtual void CvDTree::split_node_data ( CvDTreeNode * n )

protectedvirtual

Reimplemented in CvForestERTree.

virtual bool CvDTree::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 in CvBoostTree, and CvForestTree.

virtual bool CvDTree::train	(	CvMLData *	trainData,
		CvDTreeParams	params = `CvDTreeParams()`
	)

virtual

virtual bool CvDTree::train	(	CvDTreeTrainData *	trainData,
		const CvMat *	subsampleIdx
	)

virtual

Reimplemented in CvBoostTree, and CvForestTree.

virtual 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

virtual void CvDTree::try_split_node ( CvDTreeNode * n )

protectedvirtual

Reimplemented in CvBoostTree.

virtual double CvDTree::update_tree_rnc	(	int	T,
		int	fold
	)

protectedvirtual

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

virtual

Reimplemented from CvStatModel.

virtual void CvDTree::write ( CvFileStorage * fs ) const

virtual

virtual void CvDTree::write_node	(	CvFileStorage *	fs,
		CvDTreeNode *	node
	)		const

protectedvirtual

virtual void CvDTree::write_split	(	CvFileStorage *	fs,
		CvDTreeSplit *	split
	)		const

protectedvirtual

virtual void CvDTree::write_tree_nodes ( CvFileStorage * fs ) const

protectedvirtual

Friends And Related Function Documentation

friend struct cv::DTreeBestSplitFinder

friend

Member Data Documentation

CvDTreeTrainData* CvDTree::data

protected

int CvDTree::pruned_tree_idx

CvDTreeNode* CvDTree::root

protected

CvMat* CvDTree::var_importance

protected

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

ml/ml.hpp

Public Member Functions

Public Attributes

Protected Member Functions

Protected Attributes

Friends

Constructor & Destructor Documentation

Member Function Documentation

Friends And Related Function Documentation

Member Data Documentation