[← Go back to profile]

Treelite: toolbox for decision tree deployment

with Mu Li

Poster presented at SysML 2018 (2018)



This paper introduces a brand new tree library treelite. The library is a toolbox to facilitate easy deployment of models and accelerate prediction performance. It has a Python wrapper that allows users to integrate it as part of their workflow. Treelite is able to read tree ensemble models that are trained by any tree libraries, including XGBoost, LightGBM, and scikit-learn. Treelite is also designed to minimize dependencies at the time of deployment. It used to be the case that one had to ship his tree model with the original tree library that trained it; with treelite, it is no longer. Finally, treelite allows for optimizations that improve prediction performance without changing any detail of the model.

Publication Details

[← Go back to profile]