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.

Sign up for GitHub

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

Jump to bottom

Add SHAP support in Dask API #6478

Closed
hcho3 opened this issue Dec 9, 2020 · 1 comment
Closed

Add SHAP support in Dask API #6478

hcho3 opened this issue Dec 9, 2020 · 1 comment
Labels
dask

Comments

@hcho3
Copy link
Collaborator

hcho3 commented Dec 9, 2020

Currently, the predict() method in the Dask API (xgboost.dask) only supports the default prediction mode:

xgboost/python-package/xgboost/dask.py

Line 886 in 0ffaf0f

Only default prediction mode is supported right now.

It would be nice to also support SHAP prediction with Dask arrays.
@hcho3 hcho3 added the dask label Dec 9, 2020
@hcho3 hcho3 closed this as completed Dec 12, 2020
@hcho3
Copy link
Collaborator Author

hcho3 commented Dec 12, 2020

Closing, as #6117 already exposes pred_contribs=True option to the Dask API.

@hcho3 hcho3 mentioned this issue Jan 2, 2021
Closed
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
dask
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@hcho3