dask / dask/dask-examples

XGBoost example notebook uses deprecated dask-xgboost

Open
#223 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
387
Forks
225
PR merge metrics
No merged PRs in 30d

Description

This [Dask Examples notebook](https://github.com/dask/dask-examples/blob/main/machine-learning/xgboost.ipynb) uses the deprecated `dask-xgboost` rather than the native XGBoost integration.

Tweaking the notebook should be relatively straightforward by following the example code in [this blog](https://coiled.io/blog/dask-xgboost-python-example/)

```
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)
```

From the [`dask-xgboost`](https://github.com/dask/dask-xgboost) repo:

"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."

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.