|
|
|
|
|
by siboehm
1466 days ago
|
|
I built this decision tree (LightGBM) compiler last summer: https://github.com/siboehm/lleaves It get's you ~10x speedups for batch predictions, more if your model is big. It's not complicated, it ended up being <1K lines of Python code. I heard a couple of stories like yours, where people had multi-node spark clusters running LightGBM, and it always amused me because by if you compiled the trees instead you could get rid of the whole cluster. |
|
[1] https://github.com/AnotherSamWilson/miceforest