Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

XGBoost example notebook uses deprecated dask-xgboost #223

Open
avriiil opened this issue Jun 8, 2022 · 2 comments
Open

XGBoost example notebook uses deprecated dask-xgboost #223

avriiil opened this issue Jun 8, 2022 · 2 comments

Comments

@avriiil
Copy link

avriiil commented Jun 8, 2022

This Dask Examples notebook 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

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

@avriiil
Copy link
Author

avriiil commented Jun 8, 2022

I can make these changes and submit a PR

@jrbourbeau
Copy link
Member

Thanks for surfacing this @rrpelgrim. A PR that updates that example to use xgboost.dask instead of dask_xgboost would be very welcome. Let me know if you know have any questions about that process

@jrbourbeau jrbourbeau transferred this issue from dask/dask-ml Jun 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants