ControlEffect.set_data#

ControlEffect.set_data(mmm, model, X)[source]#

Update control data for new predictions.

Parameters:
mmmModel

The MMM model instance.

modelpm.Model

The cloned PyMC model used for prediction.

Xxr.Dataset or None

The dataset for prediction. Must contain "_control" if the model was trained with control columns. If None or "_control" is absent, this method does nothing.