get_prior_for_plot

pymc_marketing.mmm.summary.helpers.get_prior_for_plot(data, original_scale)

Return the correct idata group for prior predictive plotting.

Parameters:

dataMMMIDataWrapper

Wrapper holding the model’s DataTree.

original_scalebool

If True, return idata.prior (contains y_original_scale). If False, return idata.prior_predictive (contains y).

Returns:

xr.Dataset