OpenCV  2.4.13
Open Source Computer Vision
CvDTree Class Reference

#include <ml.hpp>

Inheritance diagram for CvDTree:
CvStatModel CvBoostTree CvForestTree CvForestERTree

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 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 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 ()
 
- 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 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
 
- 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: