Skip to main content

Bayesian Additive Regression Trees for Probabilistic programming with PyMC

Project description

Bayesian Additive Regression Trees for Probabilistic programming with PyMC

pymc-bart logo

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 is available on Conda-Forge. To set up a suitable Conda environment, run

conda create --name=pymc-bart --channel=conda-forge pymc-bart
conda activate pymc-bart

Alternatively, it can be installed with

pip install pymc-bart

In case you want to upgrade to 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 arXiv

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}
}

License

Apache License, Version 2.0

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

NumFOCUS

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pymc_bart-0.5.3.tar.gz (32.6 kB view hashes)

Uploaded Source

Built Distribution

pymc_bart-0.5.3-py3-none-any.whl (33.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page