blei-lab/treeffuser
Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression on tabular data with tree-based diffusion models.
Combines gradient-boosted trees with diffusion models to capture complex conditional distributions (multimodal, heteroscedastic, heavy-tailed) beyond standard Gaussian assumptions. Built on scikit-learn's API conventions for minimal configuration, it generates posterior samples that enable computing arbitrary downstream statistics like quantiles and conditional moments without manual density estimation.
Available on PyPI.
Stars
55
Forks
9
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Feb 16, 2026
Monthly downloads
43
Commits (30d)
0
Dependencies
9
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/blei-lab/treeffuser"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related models
quantgirluk/aleatory
📦 Python library for Stochastic Processes Simulation and Visualisation
yuanchenyang/smalldiffusion
Simple and readable code for training and sampling from diffusion models
TuftsBCB/RegDiffusion
Diffusion model for gene regulatory network inference.
lvyufeng/denoising-diffusion-mindspore
Implementation of Denoising Diffusion Probabilistic Model in MindSpore
NVlabs/FastGen
NVIDIA FastGen: Fast Generation from Diffusion Models