pymc-labs · juanitorduz · Jun 10, 2024 · Apr 15, 2024 · Apr 15, 2024 · Apr 16, 2024
diff --git a/data/budget_optimizer_model.nc → data/model.nc b/data/budget_optimizer_model.nc → data/model.nc
diff --git a/docs/source/notebooks/mmm/mmm_budget_allocation_example.ipynb b/docs/source/notebooks/mmm/mmm_budget_allocation_example.ipynb
diff --git a/docs/source/notebooks/mmm/mmm_example.ipynb b/docs/source/notebooks/mmm/mmm_example.ipynb
diff --git a/docs/source/notebooks/mmm/mmm_lift_test.ipynb b/docs/source/notebooks/mmm/mmm_lift_test.ipynb
diff --git a/docs/source/notebooks/mmm/mmm_tvp_example.ipynb b/docs/source/notebooks/mmm/mmm_tvp_example.ipynb
diff --git a/pymc_marketing/mmm/__init__.py b/pymc_marketing/mmm/__init__.py
@@ -12,8 +12,25 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 from pymc_marketing.mmm import base, delayed_saturated_mmm, preprocessing, validating
-from pymc_marketing.mmm.base import MMM, BaseMMM
-from pymc_marketing.mmm.delayed_saturated_mmm import DelayedSaturatedMMM
+from pymc_marketing.mmm.base import (
+    BaseValidateMMM,
+    MMMModelBuilder,
+)
+from pymc_marketing.mmm.components.adstock import (
+    AdstockTransformation,
+    DelayedAdstock,
+    GeometricAdstock,
+    WeibullAdstock,
+)
+from pymc_marketing.mmm.components.saturation import (
+    HillSaturation,
+    LogisticSaturation,
+    MichaelisMentenSaturation,
+    SaturationTransformation,
+    TanhSaturation,
+    TanhSaturationBaselined,
+)
+from pymc_marketing.mmm.delayed_saturated_mmm import MMM, DelayedSaturatedMMM
 from pymc_marketing.mmm.preprocessing import (
     preprocessing_method_X,
     preprocessing_method_y,
@@ -26,10 +43,21 @@
     "preprocessing",
     "validating",
     "MMM",
-    "BaseMMM",
+    "MMMModelBuilder",
+    "BaseValidateMMM",
     "DelayedSaturatedMMM",
     "preprocessing_method_X",
     "preprocessing_method_y",
     "validation_method_X",
     "validation_method_y",
+    "AdstockTransformation",
+    "DelayedAdstock",
+    "GeometricAdstock",
+    "WeibullAdstock",
+    "SaturationTransformation",
+    "MichaelisMentenSaturation",
+    "HillSaturation",
+    "LogisticSaturation",
+    "TanhSaturation",
+    "TanhSaturationBaselined",
 ]