-
Notifications
You must be signed in to change notification settings - Fork 325
Open
Labels
Description
Can use the example but with a single channel
import pandas as pd
from pymc_marketing.mmm import (
GeometricAdstock,
LogisticSaturation,
MMM,
)
data_url = "https://raw.githubusercontent.com/pymc-labs/pymc-marketing/main/data/mmm_example.csv"
data = pd.read_csv(data_url, parse_dates=["date_week"])
mmm = MMM(
adstock=GeometricAdstock(l_max=8),
saturation=LogisticSaturation(),
date_column="date_week",
channel_columns=["x1"],
control_columns=[
"event_1",
"event_2",
"t",
],
yearly_seasonality=2,
)
X = data.drop("y", axis=1)
y = data["y"]
mmm.fit(X, y)