CustomModelWrapper#

pydantic model pymc_marketing.mmm.budget_optimizer.CustomModelWrapper[source]#

Wrapper for the BudgetOptimizer to handle custom PyMC models.

Methods

CustomModelWrapper.__init__(base_model, ...)

Create a new model by parsing and validating input data from keyword arguments.

field adstock: Any [Optional][source]#: Default adstock placeholder with zero carryover.

field base_model: Model [Required][source]#: Underlying PyMC model to be cloned for optimization.

field channel_columns: list[str] [Required][source]#: Channel labels used for budget optimization.

field idata: InferenceData [Required][source]#