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.
Table of Contents
Installation
PyMC-BART is available on Conda-Forge. If you magange your Python dependencies and environments with Conda, this is your best option. You may also perfer to install this way if you want an easy-to-use, isolated setup in a seperate environment. This helps avoid interfering with other projects or system-wide Python installations. To set up a suitable Conda environment, run:
conda create --name=pymc-bart --channel=conda-forge pymc-bart
conda activate pymc-bart
Alternatively, you can use pip installation. This installation is generally perfered by users who use pip, Python's package installer. This is the best choice for users who are not using Conda or for those who want to install PyMC-BART into a virtual environment managed by venv or virtualenv. In this case, run:
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
Usage
Get started by using PyMC-BART to set up a BART model:
import pymc as pm
import pymc_bart as pmb
X, y = ... # Your data replaces "..."
with pm.Model() as model:
bart = pmb.BART('bart', X, y)
...
idata = pm.sample()
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{quiroga2023bayesian,
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={2023},
doi={10.48550/ARXIV.2206.03619},
archivePrefix={arXiv},
primaryClass={stat.CO}
}
License
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
2 maintainers
Metadata
2 maintainers
Release files for pymc-bart 0.13.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| pymc_bart-0.13.1.tar.gz | 34.5 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pymc_bart-0.13.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 56.7 kB
Release files / pymc_bart-0.13.1.tar.gz
| Download URL | pymc_bart-0.13.1.tar.gz |
|---|---|
| Size | 34.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
dacc7f04324a9e049ffd919e30adf56e573ca2ecd8983b548fba4b36350cb7af
|
|
BLAKE2b-256 checksum How to use checksums |
d79d2ebdf761e03abc16b1de57a2825680b3fd189cf83fb23f67d6a0f7900520
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Release files / pymc_bart-0.13.1-py3-none-any.whl
| Download URL | pymc_bart-0.13.1-py3-none-any.whl |
|---|---|
| Size | 22.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
8a4f398d3525d3c50f2f244ec4bfb886243c4b89158fbe5f8220fb0bc591fea6
|
|
BLAKE2b-256 checksum How to use checksums |
43b0176df942983f5f79b4f6bde9372a8dfc14f119e4265a3237e69f80774851
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
