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

Support gamma in GPU_Hist. (#4874)

Browse Source 
* Just prevent building the tree instead of using an explicit pruner.
This commit is contained in:
Jiaming Yuan
2019-09-24 10:16:08 +08:00
committed by GitHub
parent a40b72d127
commit 0b89cd1dfa
2 changed files with 65 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/updater_gpu_hist.cu
View File

@@ -72,8 +72,9 @@ struct ExpandEntry {
if (split.left_sum.GetHess() == 0 || split.right_sum.GetHess() == 0) {
return false;
}
if (param.max_depth > 0 && depth == param.max_depth) return false;
if (param.max_leaves > 0 && num_leaves == param.max_leaves) return false;
if (split.loss_chg < param.min_split_loss) { return false; }
if (param.max_depth > 0 && depth == param.max_depth) {return false; }
if (param.max_leaves > 0 && num_leaves == param.max_leaves) { return false; }
return true;
}