#include <ml.hpp>
Public Member Functions | |
virtual void | clear () |
CvDTreeTrainData (const CvMat *trainData, int tflag, const CvMat *responses, const CvMat *varIdx=0, const CvMat *sampleIdx=0, const CvMat *varType=0, const CvMat *missingDataMask=0, const CvDTreeParams ¶ms=CvDTreeParams(), bool _shared=false, bool _add_labels=false) | |
CvDTreeTrainData () | |
virtual void | do_responses_copy () |
virtual void | free_node (CvDTreeNode *node) |
virtual void | free_node_data (CvDTreeNode *node) |
virtual void | free_train_data () |
virtual const int * | get_cat_var_data (CvDTreeNode *n, int vi, int *cat_values_buf) |
virtual int | get_child_buf_idx (CvDTreeNode *n) |
virtual const int * | get_class_labels (CvDTreeNode *n, int *labels_buf) |
virtual const int * | get_cv_labels (CvDTreeNode *n, int *labels_buf) |
int | get_num_classes () const |
virtual const float * | get_ord_responses (CvDTreeNode *n, float *values_buf, int *sample_indices_buf) |
virtual void | get_ord_var_data (CvDTreeNode *n, int vi, float *ord_values_buf, int *sorted_indices_buf, const float **ord_values, const int **sorted_indices, int *sample_indices_buf) |
virtual const int * | get_sample_indices (CvDTreeNode *n, int *indices_buf) |
int | get_var_type (int vi) const |
virtual void | get_vectors (const CvMat *_subsample_idx, float *values, uchar *missing, float *responses, bool get_class_idx=false) |
int | get_work_var_count () const |
virtual CvDTreeNode * | new_node (CvDTreeNode *parent, int count, int storage_idx, int offset) |
virtual CvDTreeSplit * | new_split_cat (int vi, float quality) |
virtual CvDTreeSplit * | new_split_ord (int vi, float cmp_val, int split_point, int inversed, float quality) |
virtual void | read_params (CvFileStorage *fs, CvFileNode *node) |
virtual void | set_data (const CvMat *trainData, int tflag, const CvMat *responses, const CvMat *varIdx=0, const CvMat *sampleIdx=0, const CvMat *varType=0, const CvMat *missingDataMask=0, const CvDTreeParams ¶ms=CvDTreeParams(), bool _shared=false, bool _add_labels=false, bool _update_data=false) |
virtual bool | set_params (const CvDTreeParams ¶ms) |
virtual CvDTreeNode * | subsample_data (const CvMat *_subsample_idx) |
virtual void | write_params (CvFileStorage *fs) const |
virtual | ~CvDTreeTrainData () |
Public Attributes | |
CvMat * | buf |
int | buf_count |
int | buf_size |
CvMat * | cat_count |
CvMat * | cat_map |
CvMat * | cat_ofs |
int | cat_var_count |
CvMat * | counts |
CvSet * | cv_heap |
CvDTreeNode * | data_root |
CvMat * | direction |
bool | have_labels |
bool | have_priors |
int | is_buf_16u |
bool | is_classifier |
int | max_c_count |
CvSet * | node_heap |
CvSet * | nv_heap |
int | ord_var_count |
CvDTreeParams | params |
CvMat * | priors |
CvMat * | priors_mult |
const CvMat * | responses |
CvMat * | responses_copy |
cv::RNG * | rng |
int | sample_count |
bool | shared |
CvMat * | split_buf |
CvSet * | split_heap |
CvMemStorage * | temp_storage |
int | tflag |
const CvMat * | train_data |
CvMemStorage * | tree_storage |
int | var_all |
int | var_count |
CvMat * | var_idx |
CvMat * | var_type |
int | work_var_count |
CvDTreeTrainData::CvDTreeTrainData | ( | ) |
CvDTreeTrainData::CvDTreeTrainData | ( | const CvMat * | trainData, | |
int | tflag, | |||
const CvMat * | responses, | |||
const CvMat * | varIdx = 0 , |
|||
const CvMat * | sampleIdx = 0 , |
|||
const CvMat * | varType = 0 , |
|||
const CvMat * | missingDataMask = 0 , |
|||
const CvDTreeParams & | params = CvDTreeParams() , |
|||
bool | _shared = false , |
|||
bool | _add_labels = false | |||
) |
virtual CvDTreeTrainData::~CvDTreeTrainData | ( | ) | [virtual] |
virtual void CvDTreeTrainData::clear | ( | ) | [virtual] |
virtual void CvDTreeTrainData::do_responses_copy | ( | ) | [virtual] |
virtual void CvDTreeTrainData::free_node | ( | CvDTreeNode * | node | ) | [virtual] |
virtual void CvDTreeTrainData::free_node_data | ( | CvDTreeNode * | node | ) | [virtual] |
virtual void CvDTreeTrainData::free_train_data | ( | ) | [virtual] |
virtual const int* CvDTreeTrainData::get_cat_var_data | ( | CvDTreeNode * | n, | |
int | vi, | |||
int * | cat_values_buf | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
virtual int CvDTreeTrainData::get_child_buf_idx | ( | CvDTreeNode * | n | ) | [virtual] |
virtual const int* CvDTreeTrainData::get_class_labels | ( | CvDTreeNode * | n, | |
int * | labels_buf | |||
) | [virtual] |
virtual const int* CvDTreeTrainData::get_cv_labels | ( | CvDTreeNode * | n, | |
int * | labels_buf | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
int CvDTreeTrainData::get_num_classes | ( | ) | const |
virtual const float* CvDTreeTrainData::get_ord_responses | ( | CvDTreeNode * | n, | |
float * | values_buf, | |||
int * | sample_indices_buf | |||
) | [virtual] |
virtual void CvDTreeTrainData::get_ord_var_data | ( | CvDTreeNode * | n, | |
int | vi, | |||
float * | ord_values_buf, | |||
int * | sorted_indices_buf, | |||
const float ** | ord_values, | |||
const int ** | sorted_indices, | |||
int * | sample_indices_buf | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
virtual const int* CvDTreeTrainData::get_sample_indices | ( | CvDTreeNode * | n, | |
int * | indices_buf | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
int CvDTreeTrainData::get_var_type | ( | int | vi | ) | const |
virtual void CvDTreeTrainData::get_vectors | ( | const CvMat * | _subsample_idx, | |
float * | values, | |||
uchar * | missing, | |||
float * | responses, | |||
bool | get_class_idx = false | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
int CvDTreeTrainData::get_work_var_count | ( | ) | const [inline] |
virtual CvDTreeNode* CvDTreeTrainData::new_node | ( | CvDTreeNode * | parent, | |
int | count, | |||
int | storage_idx, | |||
int | offset | |||
) | [virtual] |
virtual CvDTreeSplit* CvDTreeTrainData::new_split_cat | ( | int | vi, | |
float | quality | |||
) | [virtual] |
virtual CvDTreeSplit* CvDTreeTrainData::new_split_ord | ( | int | vi, | |
float | cmp_val, | |||
int | split_point, | |||
int | inversed, | |||
float | quality | |||
) | [virtual] |
virtual void CvDTreeTrainData::read_params | ( | CvFileStorage * | fs, | |
CvFileNode * | node | |||
) | [virtual] |
virtual void CvDTreeTrainData::set_data | ( | const CvMat * | trainData, | |
int | tflag, | |||
const CvMat * | responses, | |||
const CvMat * | varIdx = 0 , |
|||
const CvMat * | sampleIdx = 0 , |
|||
const CvMat * | varType = 0 , |
|||
const CvMat * | missingDataMask = 0 , |
|||
const CvDTreeParams & | params = CvDTreeParams() , |
|||
bool | _shared = false , |
|||
bool | _add_labels = false , |
|||
bool | _update_data = false | |||
) | [virtual] |
Reimplemented in CvERTreeTrainData.
virtual bool CvDTreeTrainData::set_params | ( | const CvDTreeParams & | params | ) | [virtual] |
virtual CvDTreeNode* CvDTreeTrainData::subsample_data | ( | const CvMat * | _subsample_idx | ) | [virtual] |
Reimplemented in CvERTreeTrainData.
virtual void CvDTreeTrainData::write_params | ( | CvFileStorage * | fs | ) | const [virtual] |
const CvMat* CvDTreeTrainData::responses |