merge latest changes
This commit is contained in:
@@ -363,12 +363,12 @@ class TestDistributedGPU:
|
||||
device="cuda",
|
||||
eval_metric="error",
|
||||
n_estimators=100,
|
||||
early_stopping_rounds=early_stopping_rounds,
|
||||
)
|
||||
cls.client = local_cuda_client
|
||||
cls.fit(
|
||||
X,
|
||||
y,
|
||||
early_stopping_rounds=early_stopping_rounds,
|
||||
eval_set=[(valid_X, valid_y)],
|
||||
)
|
||||
booster = cls.get_booster()
|
||||
|
||||
Reference in New Issue
Block a user