Merge pull request #9054 from zhmu:upstream

8 years ago · 7621cc19dc
parent 4f01ad8517 3ce1bca670
commit 7621cc19dc
1 changed files with 1 additions and 1 deletions
--- a/modules/ml/src/data.cpp
+++ b/modules/ml/src/data.cpp
@ -403,7 +403,7 @@ public:
            Mat(tempCatMap).copyTo(catMap);
        }

-        if( varType.at<uchar>(ninputvars) == VAR_CATEGORICAL )
+        if( noutputvars > 0 && varType.at<uchar>(ninputvars) == VAR_CATEGORICAL )
        {
            preprocessCategorical(responses, &normCatResponses, labels, &counters, sortbuf);
            Mat(labels).copyTo(classLabels);