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);