From 15793b1301c54685ff94e5eae7f03195014ac3a1 Mon Sep 17 00:00:00 2001 From: ahb Date: Sat, 5 Dec 2015 12:48:02 +0100 Subject: [PATCH] add TestData::getTestSamples() --- modules/ml/include/opencv2/ml.hpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/modules/ml/include/opencv2/ml.hpp b/modules/ml/include/opencv2/ml.hpp index 7acce7f33c..da0cea8b9d 100644 --- a/modules/ml/include/opencv2/ml.hpp +++ b/modules/ml/include/opencv2/ml.hpp @@ -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. -- GitLab