Implement fit stump. (#8607)

2023-01-04 04:14:51 +08:00
parent 20e6087579
commit 8d545ab2a2
23 changed files with 421 additions and 60 deletions
--- a/tests/cpp/test_learner.cc
+++ b/tests/cpp/test_learner.cc
@@ -66,7 +66,7 @@ TEST(Learner, CheckGroup) {

  std::shared_ptr<DMatrix> p_mat{
      RandomDataGenerator{kNumRows, kNumCols, 0.0f}.GenerateDMatrix()};
-  std::vector<bst_float> weight(kNumGroups);
+  std::vector<bst_float> weight(kNumGroups, 1);
  std::vector<bst_int> group(kNumGroups);
  group[0] = 2;
  group[1] = 3;