search

dvbuntu / barmpy

Python module for Bayesian Additive Regression Models
https://dvbuntu.github.io/barmpy
MIT License
4 stars 0 forks source link

Implement BART in `barmpy` library #2

Open dvbuntu opened 2 months ago

dvbuntu commented 2 months ago

It'd be great to have a Python implementation of BART in barmpy! Note that BARTPy exists, but it hasn't been updated in several years. It would still serve as an excellent starting place.

This issue should also do some refactoring to barmpy.barn so generic routines can be used in both BARN and BART. That will help future features like BAR-Support Vector Machines and the like.

dvbuntu commented 2 months ago

There's also a Python BART implementation in PyMC-BART. That's a different backend (PyMC), so understanding that might be a separate but useful approach for BARN.