From e67178c69674bcc458c545e686640a1ea59c0345 Mon Sep 17 00:00:00 2001 From: Amro Date: Wed, 13 Jan 2016 19:38:49 +0200 Subject: [PATCH] fix LR predict with raw output in 2-class case In case of binary class, pred_m is initialized as zero vector and later returned when raw outputs are requested, but it is never filled. --- modules/ml/src/lr.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/ml/src/lr.cpp b/modules/ml/src/lr.cpp index a378947914..8b0a670d4e 100644 --- a/modules/ml/src/lr.cpp +++ b/modules/ml/src/lr.cpp @@ -266,6 +266,7 @@ float LogisticRegressionImpl::predict(InputArray samples, OutputArray results, i { temp_pred = calc_sigmoid(data_t*thetas.t()); CV_Assert(temp_pred.cols==1); + pred_m = temp_pred.clone(); // if greater than 0.5, predict class 0 or predict class 1 temp_pred = (temp_pred>0.5)/255;