Disable training of RTrees when CVFolds > 0

Faraadid · Dec 16, 2016 · 2585393 · 2585393
1 parent bc7cbc1
commit 2585393
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/ml/src/rtrees.cpp b/modules/ml/src/rtrees.cpp
@@ -382,6 +382,8 @@ class RTreesImpl : public RTrees
 
     bool train( const Ptr<TrainData>& trainData, int flags )
     {
+        if (impl.getCVFolds() != 0)
+            CV_Error(Error::StsBadArg, "Cross validation for RTrees is not implemented");
         return impl.train(trainData, flags);
     }