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

Merge pull request #3257 from Atanahel:gbt_bugfix_master

Browse Source
This commit is contained in:
Vadim Pisarevsky 2014-09-24 12:25:18 +00:00
commit 541b627d21
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/ml/src/gbt.cpp
View File

@ -268,6 +268,7 @@ CvGBTrees::train( const CvMat* _train_data, int _tflag,
sample_idx = cvCreateMat( 1, sample_idx_len, CV_32S ); sample_idx = cvCreateMat( 1, sample_idx_len, CV_32S );
for (int i=0; i<sample_idx_len; ++i) for (int i=0; i<sample_idx_len; ++i)
sample_idx->data.i[i] = _sample_idx->data.i[i]; sample_idx->data.i[i] = _sample_idx->data.i[i];
std::sort(sample_idx->data.i, sample_idx->data.i + sample_idx_len);
} break; } break;
case CV_8S: case CV_8S:
case CV_8U: case CV_8U:
@ -284,7 +285,6 @@ CvGBTrees::train( const CvMat* _train_data, int _tflag,
} break; } break;
default: CV_Error(CV_StsUnmatchedFormats, "_sample_idx should be a 32sC1, 8sC1 or 8uC1 vector."); default: CV_Error(CV_StsUnmatchedFormats, "_sample_idx should be a 32sC1, 8sC1 or 8uC1 vector.");
} }
std::sort(sample_idx->data.fl, sample_idx->data.fl + sample_idx_len);
} }
else else
{ {