Avoid regenerating the gradient index for approx. (#7591)

2022-01-26 21:41:30 +08:00
parent 5d7818e75d
commit e060519d4f
7 changed files with 40 additions and 18 deletions
--- a/src/tree/hist/evaluate_splits.h
+++ b/src/tree/hist/evaluate_splits.h
@@ -351,6 +351,7 @@ template <typename GradientSumT, typename ExpandEntry> class HistEvaluator {

  auto Evaluator() const { return tree_evaluator_.GetEvaluator(); }
  auto const& Stats() const { return snode_; }
+  auto Task() const { return task_; }

  float InitRoot(GradStats const& root_sum) {
    snode_.resize(1);