Probabilistic Models Python Packages

hierarchicalforecast

Probabilistic Hierarchical forecasting 👑 with statistical and econometric methods.

stochtree

Stochastic tree ensembles (BART / XBART) for supervised learning and causal inference

pydtmc

A library for discrete-time Markov chains analysis.

drforest

Fast distributional random forests with CART, MMD, and sliced-Wasserstein splitting

lymph-model

Python package for statistical modelling of lymphatic metastatic spread in head & neck cancer.

mta

Multi-Touch Attribution

calibrain

Python framework for uncertainty estimation and calibration in EEG/MEG inverse source imaging

samseg

Sequence Adaptive Multimodal SEGmentation (SAMSEG)

topomics

Topic modeling for multiomics data

adyant

Markov chain URL wordlist generator for security research and fuzzing pipelines

treeffuser

Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression on tabular data with tree-based diffusion models.

memocell

Bayesian inference of stochastic cellular processes with and without memory in Python.

deeprob-kit

A Python Library for Deep Probabilistic Modeling

ramsey

Probabilistic deep learning using JAX

torch-adf

PyTorch implementation of Assumed Density Filtering (ADF) based probabilistic neural networks

first-passage-percolation-sim

Simulations of the first passage percolation on the square lattice

keras-adf

Assumed Density Filtering (ADF) Probabilistic Networks

ppchain

Your Probabilistic Modeling Copilot