InterruptedTimeSeries.get_plot_data_bayesian#

InterruptedTimeSeries.get_plot_data_bayesian(hdi_prob=0.94, response_type='expectation')[source]#

Recover the data of the experiment along with the prediction and causal impact information.

Parameters:

  • hdi_prob (float) – Probability for which the highest density interval will be computed. The default value is defined as the default from the arviz.hdi() function.

  • response_type (Literal['expectation', 'prediction']) –

    The response type to use for predictions and impact:

    • "expectation": Uses the model expectation (μ). Excludes observation noise, focusing on the systematic causal effect.

    • "prediction": Uses the full posterior predictive (ŷ). Includes observation noise, showing the full predictive uncertainty.

Return type:

DataFrame