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

ml: moved getTestSamples implementation to src

Browse Source
This commit is contained in:
Maksim Shabunin
2016-02-12 12:32:26 +03:00
parent 15793b1301
commit 54abb83c82
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/ml/include/opencv2/ml.hpp
View File

@@ -224,11 +224,8 @@ public:
CV_WRAP virtual void setTrainTestSplitRatio(double ratio, bool shuffle=true) = 0;
CV_WRAP virtual void shuffleTrainTest() = 0;
CV_WRAP Mat getTestSamples() const {
Mat idx = getTestSampleIdx();
Mat samples = getSamples();
return idx.empty() ? Mat() : getSubVector(samples, idx);
}
/** @brief Returns matrix of test samples */
CV_WRAP Mat getTestSamples() const;
CV_WRAP static Mat getSubVector(const Mat& vec, const Mat& idx);