BudgetOptimizerWrapper.optimization_model

BudgetOptimizerWrapper.optimization_model(num_periods)

Return a PyMC model configured for num_periods optimisation steps.

Uses the wrapper’s pre-built zero data (which includes carryover periods) to produce a model consistent with the old _set_predictors_for_optimization path.

Parameters:

num_periodsint

Number of optimisation periods (accepted for backward compatibility, not used directly).

Returns:

pymc.Model

A cloned PyMC model ready for budget optimization.