StaggeredDifferenceInDifferences.get_plot_data#

StaggeredDifferenceInDifferences.get_plot_data(*, group='posterior', hdi_prob=0.94)[source]#

Get event-time plotting data.

Parameters:

  • group (Literal['prior', 'posterior']) – Which draw group to summarize. "prior" requires sample_prior_predictive(); "posterior" requires fit().

  • hdi_prob (float) – Probability for HDI interval. Only used by models carrying draws; when it differs from the value stored on the result bundle, the intervals are recomputed. Defaults to HDI_PROB (currently 0.94).

Returns:

DataFrame with event_time and att columns plus att_lower / att_upper HDI bounds (draw-carrying models) or att_std / n_obs dispersion columns (point estimates). Includes both pre-treatment (placebo) and post-treatment effects. Not cached on the experiment.

Return type:

pd.DataFrame