Bambi: A simple interface for fitting Bayesian linear models in Python

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

*Corresponding author for this work

Research output: Contribution to journalArticleScientificpeer-review

41 Citations (Scopus)
679 Downloads (Pure)

Abstract

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.
Original languageEnglish
Number of pages29
JournalJournal of Statistical Software
Volume103
Issue number15
DOIs
Publication statusPublished - 15 Aug 2022
MoE publication typeA1 Journal article-refereed

Keywords

  • Bayesian statistics
  • generalized linear models
  • multilevel modeling
  • python
  • hierarchical Bayesian modeling

Fingerprint

Dive into the research topics of 'Bambi: A simple interface for fitting Bayesian linear models in Python'. Together they form a unique fingerprint.

Cite this