From 8d1c05c36f1ec99f2ae37e2d7d2fc6cb94166f12 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Sun, 28 Nov 2021 21:46:44 -0500
Subject: [PATCH] Update to mlpack 4 conventions.

---
 .../ann/loss_functions/negative_log_likelihood_impl.hpp       | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/methods/ann/loss_functions/negative_log_likelihood_impl.hpp b/src/mlpack/methods/ann/loss_functions/negative_log_likelihood_impl.hpp
index 4ea68396dd..8fec0cd2fc 100644
--- a/src/mlpack/methods/ann/loss_functions/negative_log_likelihood_impl.hpp
+++ b/src/mlpack/methods/ann/loss_functions/negative_log_likelihood_impl.hpp
@@ -32,7 +32,7 @@ double NegativeLogLikelihood<InputDataType, OutputDataType>::Forward(
   double output = 0;
   for (size_t i = 0; i < prediction.n_cols; ++i)
   {
-    size_t currentTarget = target(i) - 1;
+    size_t currentTarget = target(i);
     Log::Assert(currentTarget < prediction.n_rows,
         "Target class out of range.");
 
@@ -52,7 +52,7 @@ void NegativeLogLikelihood<InputDataType, OutputDataType>::Backward(
   loss = arma::zeros<LossType>(prediction.n_rows, prediction.n_cols);
   for (size_t i = 0; i < prediction.n_cols; ++i)
   {
-    size_t currentTarget = target(i) - 1;
+    size_t currentTarget = target(i);
     Log::Assert(currentTarget < prediction.n_rows,
         "Target class out of range.");