Merge pull request #16017 from berak:fix_svm_train_auto

5 years ago · 41af8aa046
parent 873250f5de 0dfab6bbd0
commit 41af8aa046
1 changed files with 1 additions and 1 deletions
--- a/modules/ml/src/svm.cpp
+++ b/modules/ml/src/svm.cpp
@ -1451,7 +1451,7 @@ public:
            sortSamplesByClasses( _samples, _responses, sidx_all, class_ranges );

            //check that while cross-validation there were the samples from all the classes
-            if( class_ranges[class_count] <= 0 )
+            if ((int)class_ranges.size() < class_count + 1)
                CV_Error( CV_StsBadArg, "While cross-validation one or more of the classes have "
                "been fell out of the sample. Try to reduce <Params::k_fold>" );