Merge pull request #9054 from zhmu:upstream

7621cc19 · Alexander Alekhin · 4f01ad85 · 3ce1bca6 · 7621cc19
Commit 7621cc19 authored Jul 03, 2017 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

data.cpp modules/ml/src/data.cpp +1 -1

No files found.
--- 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);