Unify evaluation functions. (#6037)

2020-08-26 14:23:27 +08:00
parent 80c8547147
commit 2fcc4f2886
29 changed files with 570 additions and 734 deletions
--- a/tests/python-gpu/test_gpu_prediction.py
+++ b/tests/python-gpu/test_gpu_prediction.py
@@ -198,6 +198,9 @@ class TestGPUPredict(unittest.TestCase):
           tm.dataset_strategy, shap_parameter_strategy, strategies.booleans())
    @settings(deadline=None)
    def test_shap(self, num_rounds, dataset, param, all_rows):
+        if param['max_depth'] == 0 and param['max_leaves'] == 0:
+            return
+
        param.update({"predictor": "gpu_predictor", "gpu_id": 0})
        param = dataset.set_params(param)
        dmat = dataset.get_dmat()