diff --git a/src/mlpack/methods/ann/loss_functions/binary_cross_entropy_loss_impl.hpp b/src/mlpack/methods/ann/loss_functions/binary_cross_entropy_loss_impl.hpp index 5fc48d5c5a..89e7aaf1c2 100644 --- a/src/mlpack/methods/ann/loss_functions/binary_cross_entropy_loss_impl.hpp +++ b/src/mlpack/methods/ann/loss_functions/binary_cross_entropy_loss_impl.hpp @@ -35,8 +35,8 @@ BCELoss::Forward( typedef typename PredictionType::elem_type ElemType; ElemType loss = -arma::accu(target % arma::log(prediction + eps) + - (1. - target) % arma::log(1. - prediction + eps));; - if(reduction) + (1. - target) % arma::log(1. - prediction + eps)); + if (reduction) loss /= prediction.n_elem; return loss; } @@ -49,7 +49,7 @@ void BCELoss::Backward( LossType& loss) { loss = (1. - target) / (1. - prediction + eps) - target / (prediction + eps); - if(reduction) + if (reduction) loss /= prediction.n_elem; }