PiecewiseITS.get_plot_data#

PiecewiseITS.get_plot_data(*, 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:

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.

Return type:

pd.DataFrame