create_zero_dataset

pymc_marketing.mmm.utils.create_zero_dataset(model, start_date, end_date, channel_xr=None)

Create a DataFrame for future prediction, with zeros (or supplied constants).

Creates a DataFrame with dates from start_date to end_date and all model dimensions, filling channel and control columns with zeros or with values from channel_xr if provided.

If channel_xr is provided it must

• have data variables that are a subset of model.channel_columns

• be indexed only by the dimensions in model.dims (no date dimension)

The values in channel_xr are copied verbatim to the corresponding channel columns and broadcast across every date in start_date … end_date.