Use maximum category in sketch. (#7853)

2022-05-05 19:56:49 +08:00
parent 8ab5e13b5d
commit 46e0bce212
6 changed files with 117 additions and 78 deletions
--- a/tests/python-gpu/test_gpu_updaters.py
+++ b/tests/python-gpu/test_gpu_updaters.py
@@ -60,6 +60,9 @@ class TestGPUUpdaters:
    def test_categorical(self, rows, cols, rounds, cats):
        self.cputest.run_categorical_basic(rows, cols, rounds, cats, "gpu_hist")

+    def test_max_cat(self) -> None:
+        self.cputest.run_max_cat("gpu_hist")
+
    def test_categorical_32_cat(self):
        '''32 hits the bound of integer bitset, so special test'''
        rows = 1000