add TestData::getTestSamples()

pull/6103/head
ahb 9 years ago committed by Maksim Shabunin
parent 80102b76f9
commit 15793b1301
  1. 6
      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.

Loading…
Cancel
Save