Commit 54abb83c authored by Maksim Shabunin's avatar Maksim Shabunin

ml: moved getTestSamples implementation to src

parent 15793b13
...@@ -224,11 +224,8 @@ public: ...@@ -224,11 +224,8 @@ public:
CV_WRAP virtual void setTrainTestSplitRatio(double ratio, bool shuffle=true) = 0; CV_WRAP virtual void setTrainTestSplitRatio(double ratio, bool shuffle=true) = 0;
CV_WRAP virtual void shuffleTrainTest() = 0; CV_WRAP virtual void shuffleTrainTest() = 0;
CV_WRAP Mat getTestSamples() const { /** @brief Returns matrix of test samples */
Mat idx = getTestSampleIdx(); CV_WRAP Mat getTestSamples() const;
Mat samples = getSamples();
return idx.empty() ? Mat() : getSubVector(samples, idx);
}
CV_WRAP static Mat getSubVector(const Mat& vec, const Mat& idx); CV_WRAP static Mat getSubVector(const Mat& vec, const Mat& idx);
......
...@@ -50,6 +50,13 @@ static const int VAR_MISSED = VAR_ORDERED; ...@@ -50,6 +50,13 @@ static const int VAR_MISSED = VAR_ORDERED;
TrainData::~TrainData() {} 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) Mat TrainData::getSubVector(const Mat& vec, const Mat& idx)
{ {
if( idx.empty() ) if( idx.empty() )
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment