Fuse split evaluation kernels (#8026)

2022-07-05 10:24:31 +02:00
parent ff1c559084
commit 794cbaa60a
6 changed files with 308 additions and 314 deletions
--- a/tests/python-gpu/test_gpu_with_dask.py
+++ b/tests/python-gpu/test_gpu_with_dask.py
@@ -285,13 +285,15 @@ class TestDistributedGPU:
                                     'booster']
            assert hasattr(booster, 'best_score')
            dump = booster.get_dump(dump_format='json')
+            print(booster.best_iteration)
            assert len(dump) - booster.best_iteration == early_stopping_rounds + 1

            valid_X = X
            valid_y = y
            cls = dxgb.DaskXGBClassifier(objective='binary:logistic',
-                                         tree_method='gpu_hist',
-                                         n_estimators=100)
+                                         tree_method='gpu_hist', 
+                                         eval_metric='error',
+                                                                                  n_estimators=100)
            cls.client = client
            cls.fit(X, y, early_stopping_rounds=early_stopping_rounds,
                    eval_set=[(valid_X, valid_y)])