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_predictor.h
+++ b/tests/cpp/predictor/test_predictor.h
@@ -61,6 +61,8 @@ void TestInplacePrediction(dmlc::any x, std::string predictor,
                           int32_t device = -1);

 void TestPredictionWithLesserFeatures(std::string preditor_name);
+
+void TestCategoricalPrediction(std::string name);
 }  // namespace xgboost

 #endif  // XGBOOST_TEST_PREDICTOR_H_