Prepare gradient index for Quantile DMatrix. (#8103)

* Prepare gradient index for Quantile DMatrix.

- Implement push batch with adapter batch.
- Implement `GetFvalue` for prediction.
This commit is contained in:
Jiaming Yuan
2022-07-22 17:26:33 +08:00
committed by GitHub
parent 1be09848a7
commit 4a4e5c7c18
7 changed files with 254 additions and 70 deletions

View File

@@ -66,6 +66,14 @@ void TestTrainingPrediction(size_t rows, size_t bins,
learner->UpdateOneIter(i, p_hist);
}
Json model{Object{}};
learner->SaveModel(&model);
learner.reset(Learner::Create({}));
learner->LoadModel(model);
learner->SetParam("predictor", predictor);
learner->Configure();
HostDeviceVector<float> from_full;
learner->Predict(p_full, false, &from_full, 0, 0);