From 2edb998f0af41b840c8ec2e41eea8ca3507883b4 Mon Sep 17 00:00:00 2001 From: Omar Shrit Date: Wed, 28 Feb 2024 12:29:45 +0100 Subject: [PATCH] Update src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp b/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp index ba50a61d35..1029fa06af 100644 --- a/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp +++ b/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp @@ -89,8 +89,7 @@ class SSELoss gradients = (input.row(1) - input.row(0)).t(); hessians = arma::vec(input.n_cols, arma::fill::ones); - return std::pow(ApplyL1(accu(gradients)), 2) / - (accu(hessians) + lambda); + return std::pow(ApplyL1(accu(gradients)), 2) / (accu(hessians) + lambda); } private: //! The L1 regularization parameter.