From 46940109b7657a6c8a2eaa53cfbf4810b6cba1c5 Mon Sep 17 00:00:00 2001 From: Abhinav Anand Date: Mon, 4 Jan 2021 02:02:46 +0530 Subject: [PATCH] Minor code quality fix --- .../ann/loss_functions/binary_cross_entropy_loss_impl.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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; }