Tweaking the notebook should be relatively straightforward by following the example code in this blog
import xgboost as xgb
# Create the XGBoost DMatrices
dtrain = xgb.dask.DaskDMatrix(client, X_train, y_train)
dtest = xgb.dask.DaskDMatrix(client, X_test, y_test)
# train the model
output = xgb.dask.train(
client, params, dtrain, num_boost_round=4,
evals=[(dtrain, 'train')]
)
# make predictions
y_pred = xgb.dask.predict(client, output, dtest)
"Warning: Dask-XGBoost has been deprecated and is no longer maintained. The functionality of this project has been included directly in XGBoost. To use Dask and XGBoost together, please use xgboost.dask instead https://xgboost.readthedocs.io/en/latest/tutorials/dask.html."
This Dask Examples notebook uses the deprecated
dask-xgboostrather than the native XGBoost integration.Tweaking the notebook should be relatively straightforward by following the example code in this blog
From the
dask-xgboostrepo:"Warning: Dask-XGBoost has been deprecated and is no longer maintained. The functionality of this project has been included directly in XGBoost. To use Dask and XGBoost together, please use xgboost.dask instead https://xgboost.readthedocs.io/en/latest/tutorials/dask.html."