WeibullPDFAdstock#

class pymc_marketing.mmm.components.adstock.WeibullPDFAdstock(l_max=FieldInfo(annotation=NoneType, required=True, description='Maximum lag for the adstock transformation.', metadata=[Gt(gt=0)]), normalize=FieldInfo(annotation=NoneType, required=False, default=True, description='Whether to normalize the adstock values.'), mode=FieldInfo(annotation=NoneType, required=False, default=<ConvMode.After: 'After'>, description='Convolution mode.'), priors=FieldInfo(annotation=NoneType, required=False, default=None, description='Priors for the parameters.'), prefix=FieldInfo(annotation=NoneType, required=False, default=None, description='Prefix for the parameters.'))[source]#

Wrapper around weibull adstock with PDF function.

Calls pymc_marketing.mmm.transformers.weibull_adstock() with type=WeibullType.PDF and the wrapper’s l_max, normalize and mode settings.

Parameters:

lamtensor

Scale parameter of the Weibull distribution; must be positive. Default prior: Prior("Gamma", mu=2, sigma=1).

ktensor

Shape parameter of the Weibull distribution; must be positive. Default prior: Prior("Gamma", mu=3, sigma=1).

.. plot::

context:: close-figs

import matplotlib.pyplot as plt import numpy as np from pymc_marketing.mmm import WeibullPDFAdstock

rng = np.random.default_rng(0)

adstock = WeibullPDFAdstock(l_max=10) prior = adstock.sample_prior(random_seed=rng) curve = adstock.sample_curve(prior) adstock.plot_curve(curve, random_seed=rng) plt.show()

Methods

`WeibullPDFAdstock.__init__`([l_max, ...])
`WeibullPDFAdstock.apply`(x, *[, core_dim, idx])	Call within a model context.
`WeibullPDFAdstock.from_dict`(data)	Reconstruct a transformation from a dict.
`WeibullPDFAdstock.function`(x, lam, k, *, dim)	Weibull adstock function.
`WeibullPDFAdstock.plot_curve`(curve[, ...])	Plot curve HDI and samples.
`WeibullPDFAdstock.plot_curve_hdi`(curve[, ...])	Plot the HDI of the curve.
`WeibullPDFAdstock.plot_curve_samples`(curve)	Plot samples from the curve.
`WeibullPDFAdstock.sample_curve`(parameters[, ...])	Sample the adstock transformation given parameters.
`WeibullPDFAdstock.sample_prior`([coords])	Sample the priors for the transformation.
`WeibullPDFAdstock.set_dims_for_all_priors`(dims)	Set the dims for all priors.
`WeibullPDFAdstock.to_dict`([_orig])	Convert the adstock transformation to a dictionary.
`WeibullPDFAdstock.update_priors`(priors)	Update the priors for a function after initialization.
`WeibullPDFAdstock.with_default_prior_dims`(dims)	Return a copy with default prior dims (dims=None) set to `dims` instead.
`WeibullPDFAdstock.with_updated_priors`(priors)	Return a copy with updated priors.

Attributes

`WeibullPDFAdstock.combined_dims`	Get the combined dims for all the parameters.
`WeibullPDFAdstock.default_priors`
`WeibullPDFAdstock.function_priors`	Get the priors for the function.
`WeibullPDFAdstock.model_config`	Mapping from variable name to prior for the model.
`WeibullPDFAdstock.prefix`
`WeibullPDFAdstock.priors`	Get the priors for the function.
`WeibullPDFAdstock.variable_mapping`	Mapping from parameter name to variable name in the model.