ml: moved getTestSamples implementation to src

2016-02-12 12:32:26 +03:00
parent 15793b1301
commit 54abb83c82
2 changed files with 9 additions and 5 deletions
--- a/modules/ml/src/data.cpp
+++ b/modules/ml/src/data.cpp
@@ -50,6 +50,13 @@ static const int VAR_MISSED = VAR_ORDERED;

 TrainData::~TrainData() {}

+Mat TrainData::getTestSamples() const
+{
+    Mat idx = getTestSampleIdx();
+    Mat samples = getSamples();
+    return idx.empty() ? Mat() : getSubVector(samples, idx);
+}
+
 Mat TrainData::getSubVector(const Mat& vec, const Mat& idx)
 {
    if( idx.empty() )