-
Notifications
You must be signed in to change notification settings - Fork 541
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
Upgrade Treelite 1.3.0 #3855
Upgrade Treelite 1.3.0 #3855
Conversation
Will need to update Treelite in |
Blocked by rapidsai/integration#270 |
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.
Pre-approving; just need to remove one bit of testing code.
clf.convert_to_treelite_model().to_treelite_checkpoint(checkpoint_path) | ||
|
||
tl_model = treelite.Model.deserialize(checkpoint_path) |
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.
TODO: Add a convenience method to directly convert an RF object into a Treelite Model object.
The memory error in gpuCI ( |
Rerun tests |
Codecov Report
@@ Coverage Diff @@
## branch-0.20 #3855 +/- ##
===============================================
- Coverage 85.96% 85.41% -0.56%
===============================================
Files 225 227 +2
Lines 16986 17317 +331
===============================================
+ Hits 14602 14791 +189
- Misses 2384 2526 +142
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
@gpucibot merge |
This PR updates `get_treelite.cmake` to use `treelite` version `1.3.0`, similarly to #3855. Authors: - AJ Schmidt (https://github.com/ajschmidt8) Approvers: - Robert Maynard (https://github.com/robertmaynard) URL: #4014
Upgrade to Treelite 1.3.0 to take advantage of the following new features: * Faster model import for scikit-learn tree models (dmlc/treelite#264). Fixes rapidsai#3768 * Binary serializer to a file stream (dmlc/treelite#270, dmlc/treelite#273) * [EXPERIMENTAL] Add GTIL, reference inference backend (dmlc/treelite#274) Make progress towards rapidsai#3853 Depends on rapidsai/integration#270 Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - William Hicks (https://github.com/wphicks) - AJ Schmidt (https://github.com/ajschmidt8) - Dante Gama Dessavre (https://github.com/dantegd) URL: rapidsai#3855
This PR updates `get_treelite.cmake` to use `treelite` version `1.3.0`, similarly to rapidsai#3855. Authors: - AJ Schmidt (https://github.com/ajschmidt8) Approvers: - Robert Maynard (https://github.com/robertmaynard) URL: rapidsai#4014
Upgrade to Treelite 1.3.0 to take advantage of the following new features:
Make progress towards #3853
Depends on rapidsai/integration#270