create_time_varying_saturation

pymc_marketing.mmm.lift_test.create_time_varying_saturation(saturation, time_varying_var_name)

Return function and variable mapping that use a time-varying variable.

Parameters:

saturationSaturationTransformation

Any SaturationTransformation instance.

time_varying_var_namestr, optional

Name of the time-varying variable in model.

Returns:

tuple[SaturationFunc, VariableMapping]

Tuple of function and variable mapping to be used in add_saturation_observations function.