meridian.model.transformers.MediaTransformer

Contains forward and inverse media transformation methods.

Inherits From: TensorTransformer

meridian.model.transformers.MediaTransformer(
    media: meridian.backend.Tensor,
    population: meridian.backend.Tensor
)

This class stores scale factors computed from per-geo medians of the media data, normalized by the geo population.

Args
media	A tensor of dimension (n_geos, n_media_times, n_media_channels) containing the media data, used to compute the scale factors.
population	A tensor of dimension (n_geos,) containing the population of each geo, used to compute the scale factors.

Attributes
population_scaled_median_m
scale_factors_gm

Methods

forward

View source

forward(
    tensor: meridian.backend.Tensor
) -> meridian.backend.Tensor

Scales a given tensor using the stored scale factors.

inverse

View source

inverse(
    tensor: meridian.backend.Tensor
) -> meridian.backend.Tensor

Scales a given tensor using the inversed stored scale factors.