From 3eb9a655c72c089eb7889f9a587a4e5d9024d3f9 Mon Sep 17 00:00:00 2001 From: Vladislav Sovrasov Date: Thu, 7 Sep 2017 15:49:46 +0300 Subject: [PATCH] ml: disable not implemented k-fold validation in RTrees --- modules/ml/src/precomp.hpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/modules/ml/src/precomp.hpp b/modules/ml/src/precomp.hpp index 84821988b6..e8f0b4e02b 100644 --- a/modules/ml/src/precomp.hpp +++ b/modules/ml/src/precomp.hpp @@ -151,6 +151,11 @@ namespace ml CV_Error( CV_StsOutOfRange, "params.CVFolds should be =0 (the tree is not pruned) " "or n>0 (tree is pruned using n-fold cross-validation)" ); + if(val > 1) + CV_Error( CV_StsNotImplemented, + "tree pruning using cross-validation is not implemented." + "Set CVFolds to 1"); + if( val == 1 ) val = 0; CVFolds = val;