378e65a8aa8f0be4e1b9e1c1c0e2e1c85bf12271,examples/pairplot.py,,,#,11

Before Change

// Data of the Eight Schools Model
J = 8
y = np.array([28.,  8., -3.,  7., -1.,  1., 18., 12.])
sigma = np.array([15., 10., 16., 11.,  9., 11., 10., 18.])


with pm.Model() as centered_eight:
    mu = pm.Normal("mu", mu=0, sd=5)
    tau = pm.HalfCauchy("tau", beta=5)
    theta = pm.Normal("theta", mu=mu, sd=tau, shape=J)
    obs = pm.Normal("obs", mu=theta, sd=sigma, observed=y)
    centered_eight_trace = pm.sample()

az.pairplot(centered_eight_trace,
            var_names=["theta__0", "theta__1", "tau", "mu"],

After Change

centered = az.load_arviz_data("centered_eight")

coords = {"school": ["Choate", "Deerfield"]}
az.pairplot(centered, var_names=["theta", "mu"], coords=None, divergences=True)

Instances