Skip to content

pymc-devs/bartrs

Repository files navigation

BART-rs

A high-performance Rust implementation of an optimized Particle Gibbs BART (PGBART) sampler designed for performance and extensibility.

Table of Contents

Installation

PyMC-BART is available on PyPI with pre-built wheels for Linux (x86_64, aarch64), Windows (x64), and macOS (x86_64, aarch64). To install using pip

pip install pymc-bart-rs

Usage

Get started by using PyMC-BART to set up a BART model

import pymc as pm
import pymc_bartrs as pmb

X, y = ... # Your data replaces "..."
with pm.Model() as model:
    bart = pmb.BART('bart', X, y)
    ...
    idata = pm.sample()

Modifications

Issue (optional).

Clone the repository.

Fork.

Edit.

Merge request.

About

High-performance Rust-based particle Gibbs sampler for Bayesian additive regression trees

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

 
 
 

Contributors