ArviZ: Exploratory analysis of Bayesian modelsΒΆ

Build Status Coverage Status Zenodo NumFocus

ArviZ is a Python package for exploratory analysis of Bayesian models. Includes functions for posterior analysis, sample diagnostics, model checking, and comparison.

The goal is to provide backend-agnostic tools for diagnostics and visualizations of Bayesian inference in Python, by first converting inference data into xarray objects. See here for more on xarray and ArviZ.

Installation with pip is recommended

pip install arviz

ArviZ will plot NumPy arrays, dictionaries of arrays, xarray datasets, and has built-in support for PyMC3, PyStan, Pyro, and emcee objects. Support for PyMC4, TensorFlow Probability, Edward2, and Edward are on the roadmap.

Contributions and issue reports are very welcome at the github repository.

If you use ArviZ and want to cite it please use JOSS

Here is the citation in BibTeX format

@article{arviz_2019,
        title = {{ArviZ} a unified library for exploratory analysis of {Bayesian} models in {Python}},
        author = {Kumar, Ravin and Carroll, Colin and Hartikainen, Ari and Martin, Osvaldo A.},
        journal = {The Journal of Open Source Software},
        year = {2019},
        doi = {10.21105/joss.01143},
        url = {http://joss.theoj.org/papers/10.21105/joss.01143},
}