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

Small cleanup to tests. (#7585)

Browse Source 
* Use random port in dask tests to avoid warnings for occupied port.
* Increase the difficulty of AUC tests.
This commit is contained in:
Jiaming Yuan
2022-01-21 14:26:57 +08:00
committed by GitHub
parent 9fd510faa5
commit 5ddd4a9d06
2 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/python/test_eval_metrics.py
View File

@@ -177,7 +177,7 @@ class TestEvalMetrics:
"objective": "binary:logistic",
},
Xy,
num_boost_round=8,
num_boost_round=1,
)
score = booster.predict(Xy)
skl_auc = roc_auc_score(y, score)
@@ -191,7 +191,7 @@ class TestEvalMetrics:
np.testing.assert_allclose(skl_auc, auc, rtol=1e-6)
@pytest.mark.skipif(**tm.no_sklearn())
@pytest.mark.parametrize("n_samples", [100, 1000])
@pytest.mark.parametrize("n_samples", [100, 1000, 10000])
def test_roc_auc(self, n_samples):
self.run_roc_auc_binary("hist", n_samples)
@@ -229,7 +229,7 @@ class TestEvalMetrics:
"num_class": n_classes,
},
Xy,
num_boost_round=8,
num_boost_round=1,
)
score = booster.predict(Xy)
skl_auc = roc_auc_score(
@@ -248,7 +248,7 @@ class TestEvalMetrics:
np.testing.assert_allclose(skl_auc, auc, rtol=1e-5)
@pytest.mark.parametrize(
"n_samples,weighted", [(4, False), (100, False), (1000, False), (1000, True)]
"n_samples,weighted", [(4, False), (100, False), (1000, False), (10000, True)]
)
def test_roc_auc_multi(self, n_samples, weighted):
self.run_roc_auc_multi("hist", n_samples, weighted)