Linear trend (#991)

* push up some linear-trend work I had * image for the documentation * start the slopes at t=0 * additional checks at init * import at mmm module * update the image * Move to reference section * Move coords to single line * Add hierarchical trend example * add image for the example * migrate to pydantic * update images after seed * Correct the latex * fix check from tests
pymc-labs · Sep 10, 2024 · 0c10953 · 0c10953
1 parent b7c66cd
commit 0c10953
Show file tree

Hide file tree

Showing 5 changed files with 573 additions and 0 deletions.
diff --git a/docs/source/_static/hierarchical-linear-trend-prior.png b/docs/source/_static/hierarchical-linear-trend-prior.png
diff --git a/docs/source/_static/linear-trend-prior.png b/docs/source/_static/linear-trend-prior.png
diff --git a/pymc_marketing/mmm/__init__.py b/pymc_marketing/mmm/__init__.py
@@ -38,6 +38,7 @@
     saturation_from_dict,
 )
 from pymc_marketing.mmm.fourier import MonthlyFourier, YearlyFourier
+from pymc_marketing.mmm.linear_trend import LinearTrend
 from pymc_marketing.mmm.media_transformation import (
     MediaConfig,
     MediaConfigList,
@@ -85,4 +86,5 @@
     "validating",
     "validation_method_X",
     "validation_method_y",
+    "LinearTrend",
 ]