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

Persist data in dask test. (#7077)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-07-06 11:47:17 +08:00 committed by GitHub
parent b56d3d5d5c
commit ffa66aace0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/python/test_with_dask.py
View File

@ -184,6 +184,9 @@ def test_dask_predict_shape_infer(client: "Client") -> None:
def run_boost_from_prediction(
X: xgb.dask._DaskCollection, y: xgb.dask._DaskCollection, tree_method: str, client: "Client"
) -> None:
X = client.persist(X)
y = client.persist(y)
model_0 = xgb.dask.DaskXGBClassifier(
learning_rate=0.3, random_state=0, n_estimators=4,
tree_method=tree_method)