siboehm / lleaves

Compiler for LightGBM gradient-boosted trees, based on LLVM. Speeds up prediction by ≥10x.
https://lleaves.readthedocs.io/en/latest/
MIT License
347 stars 29 forks source link

Replace Deprecated Functions in `benchmark.py` Related to `treelite` #80

Open Soontosh opened 2 months ago

Soontosh commented 2 months ago

From version 4.0.0, treelite deprecated many functions and moved them to the tl2cgen package. This has resulted in fatal errors in benchmark.py, which I have fixed by replacing the affected functions with the corresponding new ones from tl2cgen.

The following links have more information: treelite GitHub Changelog 3.9.0 treelite migration guide

siboehm commented 2 months ago

thanks a lot!