HSGPPeriodic

class pymc_marketing.mmm.hsgp.HSGPPeriodic(**data)

HSGP component for periodic data.

Examples

HSGP with default configuration:

import numpy as np
import pandas as pd

import matplotlib.pyplot as plt

from pymc_marketing.mmm import HSGPPeriodic
from pymc_marketing.prior import Prior

seed = sum(map(ord, "Periodic GP"))
rng = np.random.default_rng(seed)

n = 52 * 3
dates = pd.date_range("2023-01-01", periods=n, freq="W-MON")
X = np.arange(n)
coords = {
    "time": dates,
}
scale = Prior("HalfNormal", sigma=1)
ls = Prior("InverseGamma", alpha=2, beta=1)

hsgp = HSGPPeriodic(
    scale=scale,
    m=20,
    cov_func="periodic",
    ls=ls,
    period=52,
    dims="time",
)
hsgp.register_data(X)

prior = hsgp.sample_prior(coords=coords, random_seed=rng)
curve = prior["f"]
fig, axes = hsgp.plot_curve(
    curve,
    sample_kwargs={"n": 3, "rng": rng},
)
ax = axes[0]
ax.set(xlabel="Date", ylabel="f", title="HSGP with period of 52 days")
plt.show()

(Source code, png, hires.png, pdf)

Higher dimensional HSGP with periodic data

import numpy as np
import pandas as pd

import pymc as pm

import matplotlib.pyplot as plt

from pymc_marketing.mmm import HSGPPeriodic
from pymc_marketing.prior import Prior

seed = sum(map(ord, "Higher dimensional HSGP with periodic data"))
rng = np.random.default_rng(seed)

n = 52 * 3
dates = pd.date_range("2023-01-01", periods=n, freq="W-MON")
X = np.arange(n)

scale = Prior("HalfNormal", sigma=1)
ls = Prior("InverseGamma", alpha=2, beta=1)

hsgp = HSGPPeriodic(
    X=X,
    scale=scale,
    ls=ls,
    m=20,
    cov_func="periodic",
    period=52,
    dims=("time", "channel", "product"),
)

coords = {
    "time": dates,
    "channel": ["A", "B"],
    "product": ["X", "Y", "Z"],
}
prior = hsgp.sample_prior(coords=coords, random_seed=rng)
curve = prior["f"]
fig, axes = hsgp.plot_curve(
    curve,
    sample_kwargs={"n": 3, "rng": rng},
    subplot_kwargs={"figsize": (12, 8), "ncols": 3},
)
plt.show()

(Source code, png, hires.png, pdf)

Methods

HSGPPeriodic.__init__(**data)	Create a new model by parsing and validating input data from keyword arguments.
HSGPPeriodic.construct([_fields_set])
HSGPPeriodic.copy(*[, include, exclude, ...])	Returns a copy of the model.
HSGPPeriodic.create_variable(name)	Create HSGP variable.
HSGPPeriodic.dict(*[, include, exclude, ...])
HSGPPeriodic.from_dict(data)	Create an object from a dictionary.
HSGPPeriodic.from_orm(obj)
HSGPPeriodic.json(*[, include, exclude, ...])
HSGPPeriodic.model_construct([_fields_set])	Creates a new instance of the Model class with validated data.
HSGPPeriodic.model_copy(*[, update, deep])	Usage docs: https://docs.pydantic.dev/2.10/concepts/serialization/#model_copy
HSGPPeriodic.model_dump(*[, mode, include, ...])	Usage docs: https://docs.pydantic.dev/2.10/concepts/serialization/#modelmodel_dump
HSGPPeriodic.model_dump_json(*[, indent, ...])	Usage docs: https://docs.pydantic.dev/2.10/concepts/serialization/#modelmodel_dump_json
HSGPPeriodic.model_json_schema([by_alias, ...])	Generates a JSON schema for a model class.
HSGPPeriodic.model_parametrized_name(params)	Compute the class name for parametrizations of generic classes.
HSGPPeriodic.model_post_init(_BaseModel__context)	Override this method to perform additional initialization after __init__ and model_construct.
HSGPPeriodic.model_rebuild(*[, force, ...])	Try to rebuild the pydantic-core schema for the model.
HSGPPeriodic.model_validate(obj, *[, ...])	Validate a pydantic model instance.
HSGPPeriodic.model_validate_json(json_data, *)	Usage docs: https://docs.pydantic.dev/2.10/concepts/json/#json-parsing
HSGPPeriodic.model_validate_strings(obj, *)	Validate the given object with string data against the Pydantic model.
HSGPPeriodic.parse_file(path, *[, ...])
HSGPPeriodic.parse_obj(obj)
HSGPPeriodic.parse_raw(b, *[, content_type, ...])
HSGPPeriodic.plot_curve(curve[, ...])	Plot the curve.
HSGPPeriodic.register_data(X)	Register the data to be used in the model.
HSGPPeriodic.sample_prior([coords])	Sample from the prior distribution.
HSGPPeriodic.schema([by_alias, ref_template])
HSGPPeriodic.schema_json(*[, by_alias, ...])
HSGPPeriodic.to_dict()	Convert the object to a dictionary.
HSGPPeriodic.update_forward_refs(**localns)
HSGPPeriodic.validate(value)

Attributes

model_computed_fields
model_config	Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].
model_extra	Get extra fields set during validation.
model_fields
model_fields_set	Returns the set of fields that have been explicitly set on this model instance.
ls
scale
cov_func
period
m
X
X_mid
dims