compute_waterfall_components

pymc_marketing.mmm.summary.helpers.compute_waterfall_components(data, dims=None, original_scale=True)

Mean contributions over date per component for waterfall summaries.

Matches waterfall() bar heights while preserving chain and draw for HDI summaries.

Parameters:

dataMMMIDataWrapper

Fitted model data wrapper.

dimsdict, optional

Dimension filters passed to _select_dims().

original_scalebool, default True

Whether to use original-scale contributions.

Returns:

xr.DataArray

Values with dims (chain, draw, component[, custom_dims]).