Update src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Omar Shrit
2024-02-28 12:29:45 +01:00
committed by GitHub
co-authored by Ryan Curtin
parent a9a44cad6d
commit 2edb998f0a
@@ -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.