[pyspark] Use quantile dmatrix. (#8284)

2022-10-12 20:38:53 +08:00
parent ce0382dcb0
commit 97a5b088a5
9 changed files with 225 additions and 120 deletions
--- a/tests/python-gpu/test_gpu_with_dask/test_gpu_with_dask.py
+++ b/tests/python-gpu/test_gpu_with_dask/test_gpu_with_dask.py
@@ -188,12 +188,9 @@ def run_gpu_hist(

    # See note on `ObjFunction::UpdateTreeLeaf`.
    update_leaf = dataset.name.endswith("-l1")
-    if update_leaf and len(history) == 2:
+    if update_leaf:
        assert history[0] + 1e-2 >= history[-1]
        return
-    if update_leaf and len(history) > 2:
-        assert history[0] >= history[-1]
-        return
    else:
        assert tm.non_increasing(history)