Requires setting leaf stat when expanding tree. (#5501)

* Fix GPU Hist feature importance.
2020-04-10 12:27:03 +08:00
parent dc2950fd90
commit 7d52c0b8c2
11 changed files with 179 additions and 50 deletions
--- a/src/tree/updater_gpu_hist.cu
+++ b/src/tree/updater_gpu_hist.cu
@@ -814,7 +814,8 @@ struct GPUHistMakerDevice {
    tree.ExpandNode(candidate.nid, candidate.split.findex,
                    candidate.split.fvalue, candidate.split.dir == kLeftDir,
                    base_weight, left_weight, right_weight,
-                    candidate.split.loss_chg, parent_sum.sum_hess);
+                    candidate.split.loss_chg, parent_sum.sum_hess,
+                    left_stats.GetHess(), right_stats.GetHess());
    // Set up child constraints
    node_value_constraints.resize(tree.GetNodes().size());
    node_value_constraints[candidate.nid].SetChild(