Update src/mlpack/methods/softmax_regression/softmax_regression_impl.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
2024-01-12 18:22:05 +01:00
parent f8a1936e81
commit 33de2e0812
1 changed files with 1 additions and 2 deletions
@@ -198,8 +198,7 @@ inline void SoftmaxRegression<MatType>::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);