Add base_margin for evaluation dataset. (#6591)
* Add base margin to evaluation datasets. * Unify the code base for evaluation matrices.
This commit is contained in:
@@ -314,6 +314,14 @@ class TestDistributedGPU:
|
||||
for i in range(len(ddqdm_names)):
|
||||
assert ddqdm_names[i] == dqdm_names[i]
|
||||
|
||||
sig = OrderedDict(signature(xgb.XGBRanker.fit).parameters)
|
||||
ranker_names = list(sig.keys())
|
||||
sig = OrderedDict(signature(xgb.dask.DaskXGBRanker.fit).parameters)
|
||||
dranker_names = list(sig.keys())
|
||||
|
||||
for rn, drn in zip(ranker_names, dranker_names):
|
||||
assert rn == drn
|
||||
|
||||
def run_quantile(self, name: str, local_cuda_cluster: LocalCUDACluster) -> None:
|
||||
if sys.platform.startswith("win"):
|
||||
pytest.skip("Skipping dask tests on Windows")
|
||||
|
||||
Reference in New Issue
Block a user