treelite
treelite copied to clipboard
Universal model exchange and serialization format for decision tree forests
Hi, I have a xgboost model in JSON format when I try to load it with treelite version >4.0 (`treelite.frontend.load_xgboost_model`) I get the error `Node with key 1 is orphaned...
NumPy 2.0 is coming out soon ( https://github.com/numpy/numpy/issues/24300 ). NumPy 2.0.0rc1 packages for conda & wheels came out 3 weeks back ( https://github.com/numpy/numpy/issues/24300#issuecomment-2030603395 ) To prepare for NumPy 2.0, it...
Method to export model back to sklearn. This would help in analysing/porting models better.
As discussed in #438, we are spinning off the C codegen ("compiler") part of Treelite into a separate project (dubbed `tl2cgen`). ## Goals * Keep Treelite small. It should do...
- [x] #503 - [ ] Write a tutorial for integrating Treelite into a C++ project (using CMake). - [ ] Write a tutorial for integrating Treelite into a Python...
We would like to add support for [Catboost models](https://github.com/catboost/catboost). Users of Treelite should be able to load Catboost models and run prediction. ## Overview Catboost has a custom target encoding...
When given a sparse array (`scipy.sparse.csr_matrix`), LightGBM assigns value 0 to all features that are not present in the sparse array. Currently, Treelite converts non-present features into NaNs. We will...
I noticed that my model returns different scores than the original model. I was able to boil the issue down to using a `base_score` during training. Can it be that...