From 33de2e081219046973debee434dcbbd6c2a69622 Mon Sep 17 00:00:00 2001 From: Omar Shrit Date: Fri, 12 Jan 2024 18:22:05 +0100 Subject: [PATCH] Update src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp Co-authored-by: Ryan Curtin --- .../methods/softmax_regression/softmax_regression_impl.hpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp b/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp index 1fe28b8ef7..91002383b7 100644 --- a/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp +++ b/src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp @@ -198,8 +198,7 @@ inline void SoftmaxRegression::Classify( hypothesis = arma::exp(parameters * dataset); } - probabilities = hypothesis / repmat(arma::sum(hypothesis, 0), - numClasses, 1); + probabilities = hypothesis / repmat(arma::sum(hypothesis, 0), numClasses, 1); // Prepare necessary data. labels.zeros(dataset.n_cols);