SyntheticControl.get_plot_data_bayesian#

SyntheticControl.get_plot_data_bayesian(hdi_prob=0.94, treated_unit=None)[source]#

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

Parameters:

  • hdi_prob (float) – Probability mass of the highest density interval. Defaults to the project-wide HDI_PROB.

  • treated_unit (str | None) – Which treated unit to extract data for. Must be a string name of the treated unit. If None, uses the first treated unit.

Return type:

DataFrame