ArviZ support

ArviZ support#

As of version 2.2, corner supports plotting ArviZ samples, providing a consistent interface to many MCMC samplers.

Here is a simple example of this:

import corner
import numpy as np
import arviz as az

np.random.seed(11234)

x = np.random.randn(2, 2000)
data = az.from_dict(
    posterior={"x": x, "y": np.random.randn(2, 2000, 2)},
    sample_stats={"diverging": x < -1.2},
)

figure = corner.corner(data, divergences=True)
../../_images/2d8e69e5d129f252743eff0d5514e2bba82864d2862da46609063a082d629f42.png

When using this interface, it is probably best to provide the truths parameter using a dictionary keyed on the variable names:

figure = corner.corner(data, truths={"x": 1.5, "y": [-0.3, 0.1]})
../../_images/9b1d94412156a87c014ce8cb75e98c1046cdaa96baacd97106ebc6f418d77e55.png