#include <ml.hpp>
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) |
![]() | |
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 () |
![]() | |
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) |
![]() | |
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 |
![]() | |
CvDTreeNode * | root |
CvMat * | var_importance |
CvDTreeTrainData * | data |
![]() | |
const char * | default_model_name |
Friends | |
struct | cv::ForestTreeBestSplitFinder |
Additional Inherited Members | |
![]() | |
int | pruned_tree_idx |
CvForestTree::CvForestTree | ( | ) |
|
virtual |
|
protectedvirtual |
Reimplemented from CvDTree.
|
inlinevirtual |
|
virtual |
|
virtual |
Reimplemented from CvDTree.
|
virtual |
Reimplemented from CvDTree.
|
virtual |
|
virtual |
Reimplemented from CvDTree.
|
virtual |
Reimplemented from CvDTree.
|
friend |
|
protected |