PiecewiseITS.get_plot_data#

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

Recover the data of the experiment along with prediction and effect information.

HDI columns are included only when the prediction container carries posterior draws.

Parameters:

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

  • hdi_prob (float) – Probability for the highest density interval. Defaults to HDI_PROB (currently 0.94). Ignored when the prediction container has no posterior draws.

Returns:

DataFrame containing observed data, predictions, and effects. Not cached on the experiment.

Return type:

pd.DataFrame