RegressionDiscontinuity.plot#

RegressionDiscontinuity.plot(*, round_to=2, ci_prob=0.94, hdi_prob=None, kind='ribbon', ci_kind='hdi', num_samples=50, figsize=None, show=True, legend_kwargs=None)[source]#

Plot the regression discontinuity results.

Parameters:

  • round_to (int | None) – Number of decimals used to round numerical results in the figure title (e.g. the Bayesian \(R^2\)). Defaults to 2. Use None to render raw numbers.

  • ci_prob (float) – Probability mass of the highest density interval drawn around the posterior predictive band, and the central credible interval reported in the figure title for the discontinuity at threshold. Must be in (0, 1]. Ignored for OLS models. Defaults to HDI_PROB (currently 0.94).

  • hdi_prob (float | None) – Deprecated. Use ci_prob instead.

  • kind (Literal['ribbon', 'histogram', 'spaghetti']) – How posterior uncertainty is rendered via plot_posterior_over_x(). Defaults to "ribbon". For "spaghetti", legends use draw lines rather than a shaded band. For "histogram", uncertainty is shown as a 2D density heatmap with a mean line overlay (no ribbon patch for legends).

  • ci_kind (Literal['hdi', 'eti']) – Credible interval type when kind="ribbon". Defaults to "hdi".

  • num_samples (int) – Number of posterior draws when kind="spaghetti". Defaults to 50. Ignored for other kinds.

  • figsize (tuple[float, float] | None) – Width and height of the figure in inches, passed to matplotlib.pyplot.subplots(). Defaults to None (use matplotlib’s default).

  • show (bool) – Whether to automatically display the plot. Defaults to True.

  • legend_kwargs (dict[str, Any] | None) – Keyword arguments to adjust legend placement and styling. Supported keys: loc, bbox_to_anchor, fontsize, frameon, title (bbox_transform is accepted alongside bbox_to_anchor). The existing legend is modified in place so that custom handles are preserved.

Returns:

  • fig (matplotlib.figure.Figure) – The figure that was created.

  • ax (matplotlib.axes.Axes) – The axes object containing the plot.

Return type:

tuple[Figure, Axes]