Regression Plot

See also

API Documentation: plot_lm()

Matplotlib

import matplotlib.pyplot as plt
import xarray as xr
import numpy as np
import arviz as az

az.style.use("arviz-doc")

data = az.load_arviz_data("regression1d")
x = xr.DataArray(np.linspace(0, 1, 100))
data.add_groups({"constant_data": {"x1": x}})
data.constant_data["x"] = x
data.posterior["y_model"] = (
    data.posterior["intercept"] + data.posterior["slope"] * data.constant_data["x"]
)
az.plot_lm(idata=data, y="y", x="x", y_model="y_model")

plt.show()

Download Python Source Code: matplotlib/mpl_plot_lm.py

Bokeh

import xarray as xr
import numpy as np
import arviz as az

data = az.load_arviz_data("regression1d")
x = xr.DataArray(np.linspace(0, 1, 100))
data.add_groups({"constant_data": {"x1": x}})
data.constant_data["x"] = x
data.posterior["y_model"] = (
    data.posterior["intercept"] + data.posterior["slope"] * data.constant_data["x"]
)

az.plot_lm(idata=data, y="y", x="x", y_model="y_model", backend="bokeh", figsize=(12, 6))

Download Python Source Code: bokeh/bokeh_plot_lm.py