MMMIDataWrapper.get_contributions

MMMIDataWrapper.get_contributions(original_scale=True, include_baseline=True, include_controls=True, include_seasonality=True)

Get all contribution variables in a single dataset.

Parameters:

original_scalebool, default True

Whether to return contributions in original scale

include_baselinebool, default True

Include intercept/baseline contribution

include_controlsbool, default True

Include control variable contributions (if present)

include_seasonalitybool, default True

Include seasonality contributions (if present)

Returns:

xr.Dataset

Dataset with all contribution variables

Raises:

ValueError

If original_scale=True and target_scale is not found in constant_data