CvDTree Class Reference

#include <ml.h>

Inherits CvStatModel.

Inherited by CvBoostTree, and CvForestTree.

List of all members.

Public Member Functions

 CvDTree ()
virtual ~CvDTree ()
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, CvDTreeParams params=CvDTreeParams())
virtual bool train (CvMLData *_data, CvDTreeParams _params=CvDTreeParams())
virtual float calc_error (CvMLData *_data, int type, std::vector< float > *resp=0)
virtual bool train (CvDTreeTrainData *_train_data, const CvMat *_subsample_idx)
virtual CvDTreeNodepredict (const CvMat *_sample, const CvMat *_missing_data_mask=0, bool preprocessed_input=false) 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(), CvDTreeParams params=CvDTreeParams())
virtual CvDTreeNodepredict (const cv::Mat &_sample, const cv::Mat &_missing_data_mask=cv::Mat(), bool preprocessed_input=false) const
virtual const CvMatget_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 CvDTreeNodeget_root () const
int get_pruned_tree_idx () const
CvDTreeTrainDataget_data ()
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 CvDTreeSplitfind_best_split (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

CvDTreeNoderoot
CvMatvar_importance
CvDTreeTrainDatadata
const char * default_model_name

Friends

struct cv::DTreeBestSplitFinder

Constructor & Destructor Documentation

CvDTree::CvDTree (  )
virtual CvDTree::~CvDTree (  ) [virtual]

Member Function Documentation

virtual bool CvDTree::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,
CvDTreeParams  params = CvDTreeParams() 
) [virtual]

Reimplemented in CvForestTree, and CvBoostTree.

virtual bool CvDTree::train ( CvMLData _data,
CvDTreeParams  _params = CvDTreeParams() 
) [virtual]
virtual float CvDTree::calc_error ( CvMLData _data,
int  type,
std::vector< float > *  resp = 0 
) [virtual]
virtual bool CvDTree::train ( CvDTreeTrainData _train_data,
const CvMat _subsample_idx 
) [virtual]

Reimplemented in CvForestTree, and CvBoostTree.

virtual CvDTreeNode* CvDTree::predict ( const CvMat _sample,
const CvMat _missing_data_mask = 0,
bool  preprocessed_input = false 
) const [virtual]
virtual bool CvDTree::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(),
CvDTreeParams  params = CvDTreeParams() 
) [virtual]
virtual CvDTreeNode* CvDTree::predict ( const cv::Mat _sample,
const cv::Mat _missing_data_mask = cv::Mat(),
bool  preprocessed_input = false 
) const [virtual]
virtual const CvMat* CvDTree::get_var_importance (  ) [virtual]
virtual void CvDTree::clear (  ) [virtual]

Reimplemented from CvStatModel.

Reimplemented in CvBoostTree.

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

Reimplemented from CvStatModel.

Reimplemented in CvForestTree, and CvBoostTree.

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

Reimplemented from CvStatModel.

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

Reimplemented in CvForestTree, and CvBoostTree.

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

Reimplemented in CvBoostTree.

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

Reimplemented in CvForestERTree.

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

Reimplemented in CvForestTree.

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

Reimplemented in CvForestERTree, and CvBoostTree.

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

Reimplemented in CvForestERTree, and CvBoostTree.

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

Reimplemented in CvForestERTree, and CvBoostTree.

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

Reimplemented in CvForestERTree, and CvBoostTree.

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

Reimplemented in CvBoostTree.

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

Reimplemented in CvBoostTree.

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

Reimplemented in CvForestERTree, and CvBoostTree.

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

Reimplemented in CvBoostTree.

virtual void CvDTree::prune_cv (  ) [protected, virtual]
virtual double CvDTree::update_tree_rnc ( int  T,
int  fold 
) [protected, virtual]
virtual int CvDTree::cut_tree ( int  T,
int  fold,
double  min_alpha 
) [protected, virtual]
virtual void CvDTree::free_prune_data ( bool  cut_tree ) [protected, virtual]
virtual void CvDTree::free_tree (  ) [protected, virtual]
virtual void CvDTree::write_node ( CvFileStorage fs,
CvDTreeNode node 
) const [protected, virtual]
virtual void CvDTree::write_split ( CvFileStorage fs,
CvDTreeSplit split 
) const [protected, virtual]
virtual CvDTreeNode* CvDTree::read_node ( CvFileStorage fs,
CvFileNode node,
CvDTreeNode parent 
) [protected, virtual]
virtual CvDTreeSplit* CvDTree::read_split ( CvFileStorage fs,
CvFileNode node 
) [protected, virtual]
virtual void CvDTree::write_tree_nodes ( CvFileStorage fs ) const [protected, virtual]
virtual void CvDTree::read_tree_nodes ( CvFileStorage fs,
CvFileNode node 
) [protected, virtual]
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]

Friends And Related Function Documentation

friend struct cv::DTreeBestSplitFinder [friend]

Member Data Documentation

CvDTreeNode* CvDTree::root [protected]
const char* CvStatModel::default_model_name [protected, inherited]

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