Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix typo in ElasticNet threshold function (#3527)

Browse Source
This commit is contained in:
jqmp 2018-07-29 22:08:14 -04:00 committed by Rory Mitchell
parent 2bb9b9d3db
commit dd07c25d12
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/tree/split_evaluator.cc
View File

@ -142,11 +142,12 @@ class ElasticNet final : public SplitEvaluator {
inline double ThresholdL1(double g) const { inline double ThresholdL1(double g) const {
if (g > params_.reg_alpha) { if (g > params_.reg_alpha) {
g = g - params_.reg_alpha; return g - params_.reg_alpha;
} else if (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;
} }
}; };