Add categorical data support to GPU predictor. (#6165)

2020-09-29 11:25:34 +08:00
parent 7622b8cdb8
commit 798af22ff4
6 changed files with 198 additions and 44 deletions
--- a/tests/cpp/predictor/test_gpu_predictor.cu
+++ b/tests/cpp/predictor/test_gpu_predictor.cu
@@ -221,5 +221,8 @@ TEST(GPUPredictor, Shap) {
  }
 }

+TEST(GPUPredictor, CategoricalPrediction) {
+  TestCategoricalPrediction("gpu_predictor");
+}
 }  // namespace predictor
 }  // namespace xgboost