add TestData::getTestSamples()

diff --git a/modules/ml/include/opencv2/ml.hpp b/modules/ml/include/opencv2/ml.hpp
index 7acce7f..da0cea8 100644 (file)
--- 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.