Revert perceptron_impl.hpp

eddelbuettel · Nov 22, 2024 · 5b490e4 · 5b490e4
1 parent a09ae52
commit 5b490e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlpack/methods/perceptron/perceptron_impl.hpp b/src/mlpack/methods/perceptron/perceptron_impl.hpp
@@ -322,7 +322,7 @@ void Perceptron<LearnPolicy, WeightInitializationPolicy, MatType>::Classify(
   for (size_t i = 0; i < test.n_cols; ++i)
   {
     tempLabelMat = weights.t() * test.col(i) + biases;
-    maxIndex = tempLabelMat.index_max();
+    tempLabelMat.max(maxIndex);
     predictedLabels(i) = maxIndex;
   }
 }