JakeColtman / bartpy

Bayesian Additive Regression Trees For Python
https://jakecoltman.github.io/bartpy/
MIT License
219 stars 44 forks source link

Feature Request: Converting to Cython #55

Open JackKenney opened 3 years ago

JackKenney commented 3 years ago

Since the code is written in pure python, a simple step at the end to improve performance is to compile it all using Cython. It's relatively simple to set up and the speedups are great for looped code, so it might be worth looking into.

sgbaird commented 2 years ago

Numba also worth looking at (I've had a decent amount of success with it)