YearlyFourier.aplicar

YearlyFourier.apply(dayofperiod, sum=True)

Aplicar la estacionalidad de Fourier al día del año.

Debe ser utilizado dentro de un contexto de modelo PyMC.

Parámetros:

dayofperiodXTensorLike

Día del año o día de la semana

sumbool, default True

Whether to sum the fourier contributions.

Devoluciones:

XTensorVariable

Estacionalidad de Fourier

Ejemplos

Guarde el resultado antes de sumar a través de la dimensión de prefijo.

import pandas as pd

import pymc as pm

from pymc_marketing.mmm import YearlyFourier

fourier = YearlyFourier(n_order=3)

def callback(result):
    pm.Deterministic("fourier_trend", result, dims=("date", "fourier"))

dates = pd.date_range("2023-01-01", periods=52, freq="W-MON")

coords = {
    "date": dates,
}
with pm.Model(coords=coords) as model:
    dayofyear = dates.dayofyear.to_numpy()
    fourier.apply(dayofyear, result_callback=callback)