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

[doc] [dask] fix partitioning in Dask example (#6389)

Browse Source
This commit is contained in:
James Lamb 2020-12-14 10:37:49 +00:00 committed by GitHub
parent a30461cf87
commit afc4567268
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
doc/tutorials/dask.rst
View File

@ -51,12 +51,12 @@ on a dask cluster:
num_obs = 1e5 num_obs = 1e5
num_features = 20 num_features = 20
X = da.random.random( X = da.random.random(
size=(num_obs, num_features) size=(num_obs, num_features),
chunks=(1000, num_features)
) )
y = da.random.choice( y = da.random.random(
a=[0, 1], size=(num_obs, 1),
size=num_obs, chunks=(1000, 1)
replace=True
) )
dtrain = xgb.dask.DaskDMatrix(client, X, y) dtrain = xgb.dask.DaskDMatrix(client, X, y)
@ -64,7 +64,7 @@ on a dask cluster:
output = xgb.dask.train(client, output = xgb.dask.train(client,
{'verbosity': 2, {'verbosity': 2,
'tree_method': 'hist', 'tree_method': 'hist',
'objective': 'binary:logistic' 'objective': 'reg:squarederror'
}, },
dtrain, dtrain,
num_boost_round=4, evals=[(dtrain, 'train')]) num_boost_round=4, evals=[(dtrain, 'train')])