diff --git a/modules/ml/src/lr.cpp b/modules/ml/src/lr.cpp index 988aa313ed..a378947914 100644 --- a/modules/ml/src/lr.cpp +++ b/modules/ml/src/lr.cpp @@ -303,7 +303,7 @@ float LogisticRegressionImpl::predict(InputArray samples, OutputArray results, i } } - return ( pred_labs.empty() ? 0 : pred_labs.at< int >( 0 ) ); + return ( pred_labs.empty() ? 0.f : (float) pred_labs.at< int >( 0 ) ); } Mat LogisticRegressionImpl::calc_sigmoid(const Mat& data) const