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

add TestData::getTestSamples()

Browse Source
This commit is contained in:
ahb 2015-12-05 12:48:02 +01:00 committed by Maksim Shabunin
parent 80102b76f9
commit 15793b1301
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -224,6 +224,12 @@ 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);
}
CV_WRAP static Mat getSubVector(const Mat& vec, const Mat& idx);
/** @brief Reads the dataset from a .csv file and returns the ready-to-use training data.