ml: moved getTestSamples implementation to src

pull/6103/head
Maksim Shabunin 9 years ago
parent 15793b1301
commit 54abb83c82
  1. 7
      modules/ml/include/opencv2/ml.hpp
  2. 7
      modules/ml/src/data.cpp

@ -224,11 +224,8 @@ 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);
}
/** @brief Returns matrix of test samples */
CV_WRAP Mat getTestSamples() const;
CV_WRAP static Mat getSubVector(const Mat& vec, const Mat& idx);

@ -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() )

Loading…
Cancel
Save