Support categorical split in tree model dump. (#7036)

2021-06-18 16:46:20 +08:00
parent 7968c0d051
commit 29f8fd6fee
8 changed files with 263 additions and 46 deletions
--- a/tests/python-gpu/test_gpu_with_sklearn.py
+++ b/tests/python-gpu/test_gpu_with_sklearn.py
@@ -55,7 +55,6 @@ def test_categorical():
        tree_method="gpu_hist",
        use_label_encoder=False,
        enable_categorical=True,
-        predictor="gpu_predictor",
        n_estimators=10,
    )
    X = pd.DataFrame(X.todense()).astype("category")