Journal of Statistical Software (Aug 2022)

Bambi: A Simple Interface for Fitting Bayesian Linear Models in Python

  • Tomás Capretto,
  • Camen Piho,
  • Ravin Kumar,
  • Jacob Westfall,
  • Tal Yarkoni,
  • Osvaldo A Martin

DOI
https://doi.org/10.18637/jss.v103.i15
Journal volume & issue
Vol. 103
pp. 1 – 29

Abstract

Read online

The popularity of Bayesian statistical methods has increased dramatically in recent years across many research areas and industrial applications. This is the result of a variety of methodological advances with faster and cheaper hardware as well as the development of new software tools. Here we introduce an open source Python package named Bambi (BAyesian Model Building Interface) that is built on top of the PyMC probabilistic programming framework and the ArviZ package for exploratory analysis of Bayesian models. Bambi makes it easy to specify complex generalized linear hierarchical models using a formula notation similar to those found in R. We demonstrate Bambi's versatility and ease of use with a few examples spanning a range of common statistical models including multiple regression, logistic regression, and mixed-effects modeling with crossed group specific effects. Additionally we discuss how automatic priors are constructed. Finally, we conclude with a discussion of our plans for the future development of Bambi.

Keywords