[doc] [dask] fix partitioning in Dask example (#6389)
This commit is contained in:
parent
a30461cf87
commit
afc4567268
@ -51,12 +51,12 @@ on a dask cluster:
|
||||
num_obs = 1e5
|
||||
num_features = 20
|
||||
X = da.random.random(
|
||||
size=(num_obs, num_features)
|
||||
size=(num_obs, num_features),
|
||||
chunks=(1000, num_features)
|
||||
)
|
||||
y = da.random.choice(
|
||||
a=[0, 1],
|
||||
size=num_obs,
|
||||
replace=True
|
||||
y = da.random.random(
|
||||
size=(num_obs, 1),
|
||||
chunks=(1000, 1)
|
||||
)
|
||||
|
||||
dtrain = xgb.dask.DaskDMatrix(client, X, y)
|
||||
@ -64,7 +64,7 @@ on a dask cluster:
|
||||
output = xgb.dask.train(client,
|
||||
{'verbosity': 2,
|
||||
'tree_method': 'hist',
|
||||
'objective': 'binary:logistic'
|
||||
'objective': 'reg:squarederror'
|
||||
},
|
||||
dtrain,
|
||||
num_boost_round=4, evals=[(dtrain, 'train')])
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user