generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed cpp wrappers of ML tree-based models

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

11
modules/ml/src/boost.cpp
View File

@@ -2122,9 +2122,14 @@ CvBoost::train( const Mat& _train_data, int _tflag,
const Mat& _missing_mask,
CvBoostParams _params, bool _update )
{
CvMat tdata = _train_data, responses = _responses, vidx = _var_idx,
sidx = _sample_idx, vtype = _var_type, mmask = _missing_mask;
return train(&tdata, _tflag, &responses, vidx.data.ptr ? &vidx : 0,
train_data_hdr = _train_data;
train_data_mat = _train_data;
responses_hdr = _responses;
responses_mat = _responses;
CvMat vidx = _var_idx, sidx = _sample_idx, vtype = _var_type, mmask = _missing_mask;
return train(&train_data_hdr, _tflag, &responses_hdr, vidx.data.ptr ? &vidx : 0,
sidx.data.ptr ? &sidx : 0, vtype.data.ptr ? &vtype : 0,
mmask.data.ptr ? &mmask : 0, _params, _update);
}