Update src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
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.
|
||||
|
||||
Reference in New Issue
Block a user