MMMIDataWrapper.get_contributions#

MMMIDataWrapper.get_contributions(original_scale=True, include_baseline=True, include_controls=True, include_seasonality=True)[source]#

Get all contribution variables in a single dataset.

For identity-link models, contributions are computed by multiplying log-space values by target_scale. For log-link models, a hybrid decomposition is used: counterfactual total media lift with proportional log-space channel shares.

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