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) – Prob for which the highest density interval will be computed. The default value is defined as the default from the arviz.hdi() function.

  • 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