diff --git a/src/tree/split_evaluator.cc b/src/tree/split_evaluator.cc index f30433b0d..dc9da278d 100644 --- a/src/tree/split_evaluator.cc +++ b/src/tree/split_evaluator.cc @@ -142,11 +142,12 @@ class ElasticNet final : public SplitEvaluator { inline double ThresholdL1(double g) const { if (g > params_.reg_alpha) { - g = g - params_.reg_alpha; + return g - params_.reg_alpha; } else if (g < -params_.reg_alpha) { - g = g + params_.reg_alpha; + return g + params_.reg_alpha; + } else { + return 0.0; } - return g; } };