components

Components for media transformation in the MMM model.

Examples

Use custom transformations for media in the MMM model:

from pymc_extras.prior import Prior
from pymc_marketing.mmm import (
    SaturationTransformation,
    MMM,
    WeibullPDFAdstock,
)


class InfiniteReturns(SaturationTransformation):
    def function(self, x, b):
        return b * x

    default_priors = {"b": Prior("HalfNormal")}


saturation = InfiniteReturns()
adstock = WeibullPDFAdstock(l_max=15)

mmm = MMM(
    ...,
    saturation=saturation,
    adstock=adstock,
    adstock_first=True,
)

Modules

adstock	Adstock transformations for MMM.
base	Base class for adstock and saturation functions used in MMM.
saturation	Saturation transformations for the MMM model.