-
Notifications
You must be signed in to change notification settings - Fork 540
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
Support cuML / scikit-learn RF classifiers in TreeExplainer #4447
Conversation
rerun tests |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks like this PR also includes changes from #4484. Is that intentional? If so, we'll need to wait for that to merge before this can merge.
@ajschmidt8 Yes, this is intentional, to show that the proposed changes in this PR are valid. We will need to wait until #4484 is merged. |
c1061ad
to
0a4cc3e
Compare
Undoing accidentally pushed commits that should have been directed to #4473 |
rerun tests |
1 similar comment
rerun tests |
The 2.2.0 version of Treelite incorporates the following major improvements: * dmlc/treelite#314 * dmlc/treelite#322, dmlc/treelite#327 * dmlc/treelite#325 * dmlc/treelite#332 * dmlc/treelite#330 * dmlc/treelite#333 * dmlc/treelite#334 * dmlc/treelite#304 * dmlc/treelite#335 In particular, dmlc/treelite#332, dmlc/treelite#330, dmlc/treelite#333 are required for #4447. Requires rapidsai/integration#412. EDIT. Using 2.2.1 patch release, to incorporate a hotfix (dmlc/treelite#340). Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - AJ Schmidt (https://github.com/ajschmidt8) - Dante Gama Dessavre (https://github.com/dantegd) URL: #4484
@gpucibot merge |
rerun tests |
Codecov Report
@@ Coverage Diff @@
## branch-22.02 #4447 +/- ##
===============================================
Coverage ? 85.68%
===============================================
Files ? 236
Lines ? 19334
Branches ? 0
===============================================
Hits ? 16567
Misses ? 2767
Partials ? 0
Flags with carried forward coverage won't be shown. Click here to find out more. Continue to review full report at Codecov.
|
@rapidsai/cuml-cmake-codeowners Can I get approval from one of you so that this can be merged? |
Stacked on top of #4447. Do not merge until #4447 is merged first. ~If you'd like to review before #4447 is merged, look at the net diff at hcho3#2.~ - [x] Test XGBoost models with categorical splits - [x] Test LightGBM models with categorical splits Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - Dante Gama Dessavre (https://github.com/dantegd) URL: #4473
The 2.2.0 version of Treelite incorporates the following major improvements: * dmlc/treelite#314 * dmlc/treelite#322, dmlc/treelite#327 * dmlc/treelite#325 * dmlc/treelite#332 * dmlc/treelite#330 * dmlc/treelite#333 * dmlc/treelite#334 * dmlc/treelite#304 * dmlc/treelite#335 In particular, dmlc/treelite#332, dmlc/treelite#330, dmlc/treelite#333 are required for rapidsai#4447. Requires rapidsai/integration#412. EDIT. Using 2.2.1 patch release, to incorporate a hotfix (dmlc/treelite#340). Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - AJ Schmidt (https://github.com/ajschmidt8) - Dante Gama Dessavre (https://github.com/dantegd) URL: rapidsai#4484
…#4447) Add support for: - [x] cuML RF classifiers - [x] scikit-learn RF regressors - [x] scikit-learn RF classifiers TODOs - [x] Add test cases - [x] De-duplicate path extraction logic Requires dmlc/treelite#330 Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - William Hicks (https://github.com/wphicks) - Dante Gama Dessavre (https://github.com/dantegd) URL: rapidsai#4447
Stacked on top of rapidsai#4447. Do not merge until rapidsai#4447 is merged first. ~If you'd like to review before rapidsai#4447 is merged, look at the net diff at hcho3#2.~ - [x] Test XGBoost models with categorical splits - [x] Test LightGBM models with categorical splits Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - Dante Gama Dessavre (https://github.com/dantegd) URL: rapidsai#4473
Add support for:
TODOs
Requires dmlc/treelite#330