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

[breaking] Remove the predictor param, allow fallback to prediction using DMatrix. #9129

Merged
merged 47 commits into from
Jul 3, 2023

Conversation

trivialfis
Copy link
Member

@trivialfis trivialfis commented May 6, 2023

This lowers the fallback from the sklearn interface directly to C++ gbtree. The fallback logic is clearer: when devices don't match. Along with this, the predictor parameter will be removed, and the predictor used inside xgboost will be decided solely by the booster context.

  • A DeviceOrd struct is implemented to indicate the device. It will eventually replace the gpu_id parameter.
  • The predictor parameter is removed.
  • Fallback to DMatrix when inplace_predict is not available.
  • The heuristic for choosing a predictor is only used during training.

Some of the code hasn't been updated yet, including the jupyter notebook for GPU SHAP and the benchmarking code in GPUTreesshap. I will update them once the device parameter is in place.

@trivialfis trivialfis marked this pull request as draft May 6, 2023 19:18
@trivialfis trivialfis force-pushed the inplace-predict-dmatrix branch from 60b8a75 to edba724 Compare May 31, 2023 19:59
const.

Allow inplace predict in skl.

remove configuration.

Revert "remove configuration."

This reverts commit 3b47048.

Dispatch create DMatrix.

Device name.

Dispatch based on booster context.

cleanup.

Document.

Restart removing predictor param.

Fix cpu compilation.

Use context.

Remove workaround.

Remove test.

Lint.

Remove old test.

Add note.

Add test.

fix cuda inplace predict checks.

Fix test pickling.

Revert changes.
@trivialfis trivialfis force-pushed the inplace-predict-dmatrix branch from d2ce7b1 to a93b80d Compare June 28, 2023 19:19
@trivialfis trivialfis changed the title [WIP][breaking] Remove the predictor param, allow fallback to prediction using DMatrix. [breaking] Remove the predictor param, allow fallback to prediction using DMatrix. Jun 29, 2023
@trivialfis trivialfis marked this pull request as ready for review July 1, 2023 01:24
@trivialfis trivialfis merged commit 39390cc into dmlc:master Jul 3, 2023
@trivialfis trivialfis deleted the inplace-predict-dmatrix branch July 3, 2023 11:23
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

Successfully merging this pull request may close these issues.

2 participants