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

Fix weighted samples in multi-class AUC. (#7300)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-10-11 15:12:29 +08:00
committed by GitHub
parent 69d3b1b8b4
commit 298af6f409
6 changed files with 41 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/python-gpu/test_gpu_eval_metrics.py
View File

@@ -13,9 +13,11 @@ class TestGPUEvalMetrics:
def test_roc_auc_binary(self, n_samples):
self.cpu_test.run_roc_auc_binary("gpu_hist", n_samples)
@pytest.mark.parametrize("n_samples", [4, 100, 1000])
def test_roc_auc_multi(self, n_samples):
self.cpu_test.run_roc_auc_multi("gpu_hist", n_samples)
@pytest.mark.parametrize(
"n_samples,weighted", [(4, False), (100, False), (1000, False), (1000, True)]
)
def test_roc_auc_multi(self, n_samples, weighted):
self.cpu_test.run_roc_auc_multi("gpu_hist", n_samples, weighted)
@pytest.mark.parametrize("n_samples", [4, 100, 1000])
def test_roc_auc_ltr(self, n_samples):