OptimizerCompatibleModel

class pymc_marketing.mmm.budget_optimizer.OptimizerCompatibleModel(*args, **kwargs)

Protocol for marketing mix model wrappers compatible with the BudgetOptimizer.

Any object satisfying this Protocol can be passed as the model argument to BudgetOptimizer for backward compatibility.

Attributes:

idataxarray.DataTree or arviz.InferenceData

Fitted posterior inference data. Must contain a posterior group with a channel_contribution variable (or whatever response_variable you pass).

Methods

OptimizerCompatibleModel.__init__(*args, ...)
OptimizerCompatibleModel.optimization_model(...)	Return a PyMC model configured for num_periods optimization steps.

Attributes

adstock
idata