fixed cpp wrappers of ML tree-based models

This commit is contained in:
Maria Dimashova
2013-08-23 18:13:10 +04:00
parent 5a32f0b6ca
commit 989be02562
5 changed files with 38 additions and 12 deletions

View File

@@ -942,6 +942,8 @@ protected:
CvDTreeNode* root;
CvMat* var_importance;
CvDTreeTrainData* data;
CvMat train_data_hdr, responses_hdr;
cv::Mat train_data_mat, responses_mat;
public:
int pruned_tree_idx;
@@ -1053,6 +1055,8 @@ protected:
// array of the trees of the forest
CvForestTree** trees;
CvDTreeTrainData* data;
CvMat train_data_hdr, responses_hdr;
cv::Mat train_data_mat, responses_mat;
int ntrees;
int nclasses;
double oob_error;
@@ -1268,6 +1272,8 @@ protected:
virtual void initialize_weights(double (&p)[2]);
CvDTreeTrainData* data;
CvMat train_data_hdr, responses_hdr;
cv::Mat train_data_mat, responses_mat;
CvBoostParams params;
CvSeq* weak;