Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make test for categorical data deterministic. (#8080)

Browse Source
This commit is contained in:
Jiaming Yuan 2022-07-15 14:48:39 +08:00 committed by GitHub
parent dae7a41baa
commit 647d3844dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/python-gpu/test_gpu_updaters.py
View File

@ -87,6 +87,7 @@ class TestGPUUpdaters:
def test_categorical_missing(self, rows, cols, cats): def test_categorical_missing(self, rows, cols, cats):
self.cputest.run_categorical_missing(rows, cols, cats, "gpu_hist") self.cputest.run_categorical_missing(rows, cols, cats, "gpu_hist")
@pytest.mark.skipif(**tm.no_pandas())
def test_max_cat(self) -> None: def test_max_cat(self) -> None:
self.cputest.run_max_cat("gpu_hist") self.cputest.run_max_cat("gpu_hist")

6
tests/python/test_updaters.py
View File

@ -205,10 +205,13 @@ class TestTreeMethod:
def run_max_cat(self, tree_method: str) -> None: def run_max_cat(self, tree_method: str) -> None:
"""Test data with size smaller than number of categories.""" """Test data with size smaller than number of categories."""
import pandas as pd import pandas as pd
rng = np.random.default_rng(0)
n_cat = 100 n_cat = 100
n = 5 n = 5
X = pd.Series( X = pd.Series(
["".join(choice(ascii_lowercase) for i in range(3)) for i in range(n_cat)], ["".join(rng.choice(list(ascii_lowercase), size=3)) for i in range(n_cat)],
dtype="category", dtype="category",
)[:n].to_frame() )[:n].to_frame()
@ -222,6 +225,7 @@ class TestTreeMethod:
assert tm.non_increasing(reg.evals_result()["validation_0"]["rmse"]) assert tm.non_increasing(reg.evals_result()["validation_0"]["rmse"])
@pytest.mark.parametrize("tree_method", ["hist", "approx"]) @pytest.mark.parametrize("tree_method", ["hist", "approx"])
@pytest.mark.skipif(**tm.no_pandas())
def test_max_cat(self, tree_method) -> None: def test_max_cat(self, tree_method) -> None:
self.run_max_cat(tree_method) self.run_max_cat(tree_method)