From 093c4433f13d54718e893ee3fc81e3ec0f1bdc5f Mon Sep 17 00:00:00 2001 From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com> Date: Wed, 30 Jun 2021 23:02:53 +0530 Subject: [PATCH] Fixed filling the hessian vector with 1 --- src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp b/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp index fc4f941620..c754e9e7eb 100644 --- a/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp +++ b/src/mlpack/methods/xgboost/loss_functions/sse_loss.hpp @@ -75,7 +75,7 @@ class SSELoss arma::is_Row::value>> VecType Hessians(const VecType& /* observed */, const VecType& values) { - VecType h(values.n_elem, 1); + VecType h(values.n_elem, arma::fill::ones); return h; }