siboehm / lleaves

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

add function_name to compiler #21

Closed fuyw closed 2 years ago

fuyw commented 2 years ago

Add support to custom function_name in compilation.

siboehm commented 2 years ago

Sorry for taking so long to have a look at this! I'll be adding a few commits to clean things up and setup testing, then I'll merge & release.

siboehm commented 2 years ago

Closes #19

siboehm commented 2 years ago

Thanks for bringing up the issue and submitting a fix @fuyw !

fuyw commented 2 years ago

Thanks for the nice repo.