Accept iterator in device dmatrix. (#5783)

* Remove Device DMatrix.
2020-07-07 21:44:48 +08:00
parent 048d969be4
commit a3ec964346
12 changed files with 495 additions and 343 deletions
--- a/tests/cpp/predictor/test_predictor.cc
+++ b/tests/cpp/predictor/test_predictor.cc
@@ -54,6 +54,7 @@ void TestTrainingPrediction(size_t rows, size_t bins,
    learner->SetParam("objective", "multi:softprob");
    learner->SetParam("num_feature", std::to_string(kCols));
    learner->SetParam("num_class", std::to_string(kClasses));
+    learner->SetParam("max_bin", std::to_string(bins));
    learner->Configure();

    for (size_t i = 0; i < kIters; ++i) {