#include <ml.h>
Inherits CvDTree.
Public Member Functions | |
CvBoostTree () | |
virtual | ~CvBoostTree () |
virtual bool | train (CvDTreeTrainData *_train_data, const CvMat *subsample_idx, CvBoost *ensemble) |
virtual void | scale (double s) |
virtual void | read (CvFileStorage *fs, CvFileNode *node, CvBoost *ensemble, CvDTreeTrainData *_data) |
virtual void | clear () |
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 (CvDTreeTrainData *_train_data, const CvMat *_subsample_idx) |
virtual void | read (CvFileStorage *fs, CvFileNode *node) |
virtual void | read (CvFileStorage *fs, CvFileNode *node, CvDTreeTrainData *data) |
virtual bool | train (CvMLData *_data, CvDTreeParams _params=CvDTreeParams()) |
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 float | calc_error (CvMLData *_data, int type, std::vector< float > *resp=0) |
virtual CvDTreeNode * | predict (const CvMat *_sample, const CvMat *_missing_data_mask=0, bool preprocessed_input=false) const |
virtual CvDTreeNode * | predict (const cv::Mat &_sample, const cv::Mat &_missing_data_mask=cv::Mat(), bool preprocessed_input=false) const |
virtual const CvMat * | get_var_importance () |
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 () |
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 void | try_split_node (CvDTreeNode *n) |
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 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 void | calc_node_value (CvDTreeNode *n) |
virtual double | calc_node_dir (CvDTreeNode *n) |
virtual bool | do_train (const CvMat *_subsample_idx) |
virtual void | split_node_data (CvDTreeNode *n) |
virtual CvDTreeSplit * | find_best_split (CvDTreeNode *n) |
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 | 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 | |
CvBoost * | ensemble |
CvDTreeNode * | root |
CvMat * | var_importance |
CvDTreeTrainData * | data |
const char * | default_model_name |
Friends | |
struct | cv::DTreeBestSplitFinder |
CvBoostTree::CvBoostTree | ( | ) |
virtual CvBoostTree::~CvBoostTree | ( | ) | [virtual] |
virtual bool CvBoostTree::train | ( | CvDTreeTrainData * | _train_data, |
const CvMat * | subsample_idx, | ||
CvBoost * | ensemble | ||
) | [virtual] |
virtual void CvBoostTree::scale | ( | double | s ) | [virtual] |
virtual void CvBoostTree::read | ( | CvFileStorage * | fs, |
CvFileNode * | node, | ||
CvBoost * | ensemble, | ||
CvDTreeTrainData * | _data | ||
) | [virtual] |
virtual void CvBoostTree::clear | ( | ) | [virtual] |
Reimplemented from CvDTree.
virtual bool CvBoostTree::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 from CvDTree.
virtual bool CvBoostTree::train | ( | CvDTreeTrainData * | _train_data, |
const CvMat * | _subsample_idx | ||
) | [virtual] |
Reimplemented from CvDTree.
virtual void CvBoostTree::read | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [virtual] |
Reimplemented from CvDTree.
virtual void CvBoostTree::read | ( | CvFileStorage * | fs, |
CvFileNode * | node, | ||
CvDTreeTrainData * | data | ||
) | [virtual] |
Reimplemented from CvDTree.
virtual void CvBoostTree::try_split_node | ( | CvDTreeNode * | n ) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_surrogate_split_ord | ( | CvDTreeNode * | n, |
int | vi, | ||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_surrogate_split_cat | ( | CvDTreeNode * | n, |
int | vi, | ||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_split_ord_class | ( | CvDTreeNode * | n, |
int | vi, | ||
float | init_quality = 0 , |
||
CvDTreeSplit * | _split = 0 , |
||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_split_cat_class | ( | CvDTreeNode * | n, |
int | vi, | ||
float | init_quality = 0 , |
||
CvDTreeSplit * | _split = 0 , |
||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_split_ord_reg | ( | CvDTreeNode * | n, |
int | vi, | ||
float | init_quality = 0 , |
||
CvDTreeSplit * | _split = 0 , |
||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual CvDTreeSplit* CvBoostTree::find_split_cat_reg | ( | CvDTreeNode * | n, |
int | vi, | ||
float | init_quality = 0 , |
||
CvDTreeSplit * | _split = 0 , |
||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented from CvDTree.
virtual void CvBoostTree::calc_node_value | ( | CvDTreeNode * | n ) | [protected, virtual] |
Reimplemented from CvDTree.
virtual double CvBoostTree::calc_node_dir | ( | CvDTreeNode * | n ) | [protected, virtual] |
Reimplemented from CvDTree.
virtual bool CvDTree::train | ( | CvMLData * | _data, |
CvDTreeParams | _params = CvDTreeParams() |
||
) | [virtual, inherited] |
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, inherited] |
virtual float CvDTree::calc_error | ( | CvMLData * | _data, |
int | type, | ||
std::vector< float > * | resp = 0 |
||
) | [virtual, inherited] |
virtual CvDTreeNode* CvDTree::predict | ( | const CvMat * | _sample, |
const CvMat * | _missing_data_mask = 0 , |
||
bool | preprocessed_input = false |
||
) | const [virtual, inherited] |
virtual CvDTreeNode* CvDTree::predict | ( | const cv::Mat & | _sample, |
const cv::Mat & | _missing_data_mask = cv::Mat() , |
||
bool | preprocessed_input = false |
||
) | const [virtual, inherited] |
virtual const CvMat* CvDTree::get_var_importance | ( | ) | [virtual, inherited] |
virtual void CvDTree::write | ( | CvFileStorage * | fs, |
const char * | name | ||
) | const [virtual, inherited] |
Reimplemented from CvStatModel.
virtual void CvDTree::write | ( | CvFileStorage * | fs ) | const [virtual, inherited] |
const CvDTreeNode* CvDTree::get_root | ( | ) | const [inherited] |
int CvDTree::get_pruned_tree_idx | ( | ) | const [inherited] |
CvDTreeTrainData* CvDTree::get_data | ( | ) | [inherited] |
virtual bool CvDTree::do_train | ( | const CvMat * | _subsample_idx ) | [protected, virtual, inherited] |
virtual void CvDTree::split_node_data | ( | CvDTreeNode * | n ) | [protected, virtual, inherited] |
Reimplemented in CvForestERTree.
virtual CvDTreeSplit* CvDTree::find_best_split | ( | CvDTreeNode * | n ) | [protected, virtual, inherited] |
Reimplemented in CvForestTree.
virtual void CvDTree::complete_node_dir | ( | CvDTreeNode * | node ) | [protected, virtual, inherited] |
virtual void CvDTree::cluster_categories | ( | const int * | vectors, |
int | vector_count, | ||
int | var_count, | ||
int * | sums, | ||
int | k, | ||
int * | cluster_labels | ||
) | [protected, virtual, inherited] |
virtual void CvDTree::prune_cv | ( | ) | [protected, virtual, inherited] |
virtual double CvDTree::update_tree_rnc | ( | int | T, |
int | fold | ||
) | [protected, virtual, inherited] |
virtual int CvDTree::cut_tree | ( | int | T, |
int | fold, | ||
double | min_alpha | ||
) | [protected, virtual, inherited] |
virtual void CvDTree::free_prune_data | ( | bool | cut_tree ) | [protected, virtual, inherited] |
virtual void CvDTree::free_tree | ( | ) | [protected, virtual, inherited] |
virtual void CvDTree::write_node | ( | CvFileStorage * | fs, |
CvDTreeNode * | node | ||
) | const [protected, virtual, inherited] |
virtual void CvDTree::write_split | ( | CvFileStorage * | fs, |
CvDTreeSplit * | split | ||
) | const [protected, virtual, inherited] |
virtual CvDTreeNode* CvDTree::read_node | ( | CvFileStorage * | fs, |
CvFileNode * | node, | ||
CvDTreeNode * | parent | ||
) | [protected, virtual, inherited] |
virtual CvDTreeSplit* CvDTree::read_split | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [protected, virtual, inherited] |
virtual void CvDTree::write_tree_nodes | ( | CvFileStorage * | fs ) | const [protected, virtual, inherited] |
virtual void CvDTree::read_tree_nodes | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [protected, virtual, inherited] |
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] |
friend struct cv::DTreeBestSplitFinder [friend, inherited] |
CvBoost* CvBoostTree::ensemble [protected] |
CvDTreeNode* CvDTree::root [protected, inherited] |
CvMat* CvDTree::var_importance [protected, inherited] |
CvDTreeTrainData* CvDTree::data [protected, inherited] |
int CvDTree::pruned_tree_idx [inherited] |
const char* CvStatModel::default_model_name [protected, inherited] |