Implement contribution prediction with QuantileDMatrix (#10043)

--------- Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-02-19 14:03:29 +01:00
parent 057f03cacc
commit edf501d227
6 changed files with 137 additions and 55 deletions
--- a/tests/cpp/predictor/test_cpu_predictor.cc
+++ b/tests/cpp/predictor/test_cpu_predictor.cc
@@ -148,7 +148,7 @@ TEST(CPUPredictor, GHistIndexTraining) {
  auto adapter = data::ArrayAdapter(columnar.c_str());
  std::shared_ptr<DMatrix> p_full{
      DMatrix::Create(&adapter, std::numeric_limits<float>::quiet_NaN(), 1)};
-  TestTrainingPrediction(&ctx, kRows, kBins, p_full, p_hist);
+  TestTrainingPrediction(&ctx, kRows, kBins, p_full, p_hist, true);
 }

 TEST(CPUPredictor, CategoricalPrediction) {