pymc-bart 0.1.0

Bayesian Additive Regression Trees for Probabilistic programming with PyMC

Bayesian Additive Regression Trees for Probabilistic programming with PyMC

PyMC-BART extends PyMC probabilistic programming framework to be able to define and solve models including a BART random variable. PyMC-BART also includes a few helpers function to aid with the interpretation of those models and perform variable selection.

Installation

PyMC-BART requires a working Python interpreter (3.8+). We recommend installing Python and key numerical libraries using the Anaconda Distribution, which has one-click installers available on all major platforms.

Assuming a standard Python environment is installed on your machine (including pip), PyMC-BART itself can be installed in one line using pip:

pip install pymc-bart

Alternatively, if you want the bleeding edge version of the package you can install from GitHub:

pip install git+https://github.com/pymc-devs/pymc-bart.git

Contributions

PyMC-BART is a community project and welcomes contributions. Additional information can be found in the Contributing Readme

Code of Conduct

PyMC-BART wishes to maintain a positive community. Additional details can be found in the Code of Conduct

Citation

If you use PyMC-BART and want to cite it please use

Here is the citation in BibTeX format

@misc{quiroga2022bart,
title={Bayesian additive regression trees for probabilistic programming},
author={Quiroga, Miriana and Garay, Pablo G and Alonso, Juan M. and Loyola, Juan Martin and Martin, Osvaldo A},
year={2022},
doi={10.48550/ARXIV.2206.03619},
archivePrefix={arXiv},
primaryClass={stat.CO}
}

Donations

PyMC-BART , as other pymc-devs projects, is a non-profit project under the NumFOCUS umbrella. If you want to support PyMC-BART financially, you can donate here.

Sponsors