#include <ml.hpp>
Public Member Functions | |
virtual float | calc_error (CvMLData *trainData, int type, std::vector< float > *resp=0) |
virtual CV_WRAP void | clear () |
CV_WRAP | CvDTree () |
CvDTreeTrainData * | get_data () |
int | get_pruned_tree_idx () const |
const CvDTreeNode * | get_root () const |
virtual const CvMat * | get_var_importance () |
virtual CV_WRAP cv::Mat | getVarImportance () |
virtual CV_WRAP CvDTreeNode * | predict (const cv::Mat &sample, const cv::Mat &missingDataMask=cv::Mat(), bool preprocessedInput=false) const |
virtual CvDTreeNode * | predict (const CvMat *sample, const CvMat *missingDataMask=0, bool preprocessedInput=false) const |
virtual void | read (CvFileStorage *fs, CvFileNode *node) |
virtual void | read (CvFileStorage *fs, CvFileNode *node, CvDTreeTrainData *data) |
virtual bool | train (CvDTreeTrainData *trainData, const CvMat *subsampleIdx) |
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 CV_WRAP 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 void | write (CvFileStorage *fs) const |
virtual void | write (CvFileStorage *fs, const char *name) const |
virtual | ~CvDTree () |
Public Attributes | |
int | pruned_tree_idx |
Protected Member Functions | |
virtual double | calc_node_dir (CvDTreeNode *node) |
virtual void | calc_node_value (CvDTreeNode *node) |
virtual void | cluster_categories (const int *vectors, int vector_count, int var_count, int *sums, int k, int *cluster_labels) |
virtual void | complete_node_dir (CvDTreeNode *node) |
virtual int | cut_tree (int T, int fold, double min_alpha) |
virtual bool | do_train (const CvMat *_subsample_idx) |
virtual CvDTreeSplit * | find_best_split (CvDTreeNode *n) |
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_cat_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, 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_ord_reg (CvDTreeNode *n, int vi, float init_quality=0, CvDTreeSplit *_split=0, uchar *ext_buf=0) |
virtual CvDTreeSplit * | find_surrogate_split_cat (CvDTreeNode *n, int vi, uchar *ext_buf=0) |
virtual CvDTreeSplit * | find_surrogate_split_ord (CvDTreeNode *n, int vi, uchar *ext_buf=0) |
virtual void | free_prune_data (bool cut_tree) |
virtual void | free_tree () |
virtual void | prune_cv () |
virtual CvDTreeNode * | read_node (CvFileStorage *fs, CvFileNode *node, CvDTreeNode *parent) |
virtual CvDTreeSplit * | read_split (CvFileStorage *fs, CvFileNode *node) |
virtual void | read_tree_nodes (CvFileStorage *fs, CvFileNode *node) |
virtual void | split_node_data (CvDTreeNode *n) |
virtual void | try_split_node (CvDTreeNode *n) |
virtual double | update_tree_rnc (int T, int fold) |
virtual void | write_node (CvFileStorage *fs, CvDTreeNode *node) const |
virtual void | write_split (CvFileStorage *fs, CvDTreeSplit *split) const |
virtual void | write_tree_nodes (CvFileStorage *fs) const |
Protected Attributes | |
CvDTreeTrainData * | data |
CvDTreeNode * | root |
CvMat * | var_importance |
Friends | |
struct | cv::DTreeBestSplitFinder |
CV_WRAP CvDTree::CvDTree | ( | ) |
virtual CvDTree::~CvDTree | ( | ) | [virtual] |
virtual float CvDTree::calc_error | ( | CvMLData * | trainData, |
int | type, | ||
std::vector< float > * | resp = 0 |
||
) | [virtual] |
virtual double CvDTree::calc_node_dir | ( | CvDTreeNode * | node ) | [protected, virtual] |
Reimplemented in CvForestERTree, and CvBoostTree.
virtual void CvDTree::calc_node_value | ( | CvDTreeNode * | node ) | [protected, virtual] |
Reimplemented in CvBoostTree.
virtual CV_WRAP 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 | ||
) | [protected, virtual] |
virtual void CvDTree::complete_node_dir | ( | CvDTreeNode * | node ) | [protected, virtual] |
virtual int CvDTree::cut_tree | ( | int | T, |
int | fold, | ||
double | min_alpha | ||
) | [protected, virtual] |
virtual bool CvDTree::do_train | ( | const CvMat * | _subsample_idx ) | [protected, virtual] |
virtual CvDTreeSplit* CvDTree::find_best_split | ( | CvDTreeNode * | n ) | [protected, virtual] |
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 |
||
) | [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_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_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_surrogate_split_cat | ( | CvDTreeNode * | n, |
int | vi, | ||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented in CvBoostTree.
virtual CvDTreeSplit* CvDTree::find_surrogate_split_ord | ( | CvDTreeNode * | n, |
int | vi, | ||
uchar * | ext_buf = 0 |
||
) | [protected, virtual] |
Reimplemented in CvBoostTree.
virtual void CvDTree::free_prune_data | ( | bool | cut_tree ) | [protected, virtual] |
virtual void CvDTree::free_tree | ( | ) | [protected, virtual] |
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_WRAP cv::Mat CvDTree::getVarImportance | ( | ) | [virtual] |
virtual CvDTreeNode* CvDTree::predict | ( | const CvMat * | sample, |
const CvMat * | missingDataMask = 0 , |
||
bool | preprocessedInput = false |
||
) | const [virtual] |
virtual CV_WRAP CvDTreeNode* CvDTree::predict | ( | const cv::Mat & | sample, |
const cv::Mat & | missingDataMask = cv::Mat() , |
||
bool | preprocessedInput = false |
||
) | const [virtual] |
virtual void CvDTree::prune_cv | ( | ) | [protected, virtual] |
virtual void CvDTree::read | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [virtual] |
Reimplemented from CvStatModel.
Reimplemented in CvForestTree, and CvBoostTree.
virtual void CvDTree::read | ( | CvFileStorage * | fs, |
CvFileNode * | node, | ||
CvDTreeTrainData * | data | ||
) | [virtual] |
Reimplemented in CvForestTree, and CvBoostTree.
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::read_tree_nodes | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [protected, virtual] |
virtual void CvDTree::split_node_data | ( | CvDTreeNode * | n ) | [protected, virtual] |
Reimplemented in CvForestERTree.
virtual bool CvDTree::train | ( | CvMLData * | trainData, |
CvDTreeParams | params = CvDTreeParams() |
||
) | [virtual] |
virtual bool CvDTree::train | ( | CvDTreeTrainData * | trainData, |
const CvMat * | subsampleIdx | ||
) | [virtual] |
Reimplemented in CvForestTree, and CvBoostTree.
virtual CV_WRAP 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 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 CvForestTree, and CvBoostTree.
virtual void CvDTree::try_split_node | ( | CvDTreeNode * | n ) | [protected, virtual] |
Reimplemented in CvBoostTree.
virtual double CvDTree::update_tree_rnc | ( | int | T, |
int | fold | ||
) | [protected, virtual] |
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 [protected, virtual] |
virtual void CvDTree::write_split | ( | CvFileStorage * | fs, |
CvDTreeSplit * | split | ||
) | const [protected, virtual] |
virtual void CvDTree::write_tree_nodes | ( | CvFileStorage * | fs ) | const [protected, virtual] |
friend struct cv::DTreeBestSplitFinder [friend] |
CvDTreeTrainData* CvDTree::data [protected] |
CvDTreeNode* CvDTree::root [protected] |
CvMat* CvDTree::var_importance [protected] |