Merge pull request opencv#5127 from LorenaGdL:data.cpp-(ml-module)

vpisarev · vpisarev · commit b6869df55670 · 2015-08-25T07:42:41.000Z
diff --git a/modules/ml/src/data.cpp b/modules/ml/src/data.cpp
@@ -636,9 +636,18 @@ class TrainDataImpl : public TrainData
                 vtypes[ninputvars] = VAR_CATEGORICAL;
         }
 
-        Mat(nsamples, noutputvars, CV_32F, &allresponses[0]).copyTo(tempResponses);
-        setData(tempSamples, ROW_SAMPLE, tempResponses, noArray(), noArray(),
-                noArray(), Mat(vtypes).clone(), tempMissing);
+        //If there are responses in the csv file, save them. If not, responses matrix will contain just zeros
+        if (noutputvars != 0){
+            Mat(nsamples, noutputvars, CV_32F, &allresponses[0]).copyTo(tempResponses);
+            setData(tempSamples, ROW_SAMPLE, tempResponses, noArray(), noArray(),
+                    noArray(), Mat(vtypes).clone(), tempMissing);
+        }
+        else{
+            Mat zero_mat(nsamples, 1, CV_32F, Scalar(0));
+            zero_mat.copyTo(tempResponses);
+            setData(tempSamples, ROW_SAMPLE, tempResponses, noArray(), noArray(),
+                    noArray(), noArray(), tempMissing);
+        }
         bool ok = !samples.empty();
         if(ok)
             std::swap(tempNameMap, nameMap);
