Disable training of RTrees when CVFolds > 0

d2e5bea1 · Vladislav Sovrasov · 75eeb25c · d2e5bea1
Commit d2e5bea1 authored Dec 16, 2016 by Vladislav Sovrasov
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

rtrees.cpp modules/ml/src/rtrees.cpp +2 -0

No files found.
--- a/modules/ml/src/rtrees.cpp
+++ b/modules/ml/src/rtrees.cpp
@@ -382,6 +382,8 @@ public:

    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);
    }