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::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::Backward( loss = arma::zeros(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.");