Accept histogram cut instead gradient index in evaluation. (#7336)

2021-10-20 18:04:46 +08:00
parent 15685996fc
commit 8d7c6366d7
4 changed files with 13 additions and 11 deletions
--- a/tests/cpp/tree/hist/test_evaluate_splits.cc
+++ b/tests/cpp/tree/hist/test_evaluate_splits.cc
@@ -58,7 +58,7 @@ template <typename GradientSumT> void TestEvaluateSplits() {
  entries.front().depth = 0;

  evaluator.InitRoot(GradStats{total_gpair});
-  evaluator.EvaluateSplits(hist, gmat, tree, &entries);
+  evaluator.EvaluateSplits(hist, gmat.cut, tree, &entries);

  auto best_loss_chg =
      evaluator.Evaluator().CalcSplitGain(