Support column split in multi-target hist (#9171)

2023-05-26 01:56:05 -07:00
parent acd363033e
commit 5b69534b43
17 changed files with 386 additions and 96 deletions
--- a/tests/cpp/tree/test_constraints.cc
+++ b/tests/cpp/tree/test_constraints.cc
@@ -79,7 +79,7 @@ TEST(CPUMonoConstraint, Basic) {
  auto Xy = RandomDataGenerator{kRows, kCols, 0.0}.GenerateDMatrix(true);
  auto sampler = std::make_shared<common::ColumnSampler>();

-  HistEvaluator<CPUExpandEntry> evalutor{&ctx, &param, Xy->Info(), sampler};
+  HistEvaluator evalutor{&ctx, &param, Xy->Info(), sampler};
  evalutor.InitRoot(GradStats{2.0, 2.0});

  SplitEntry split;