Change to beta_geo model doc, where mode_config had incorrect diction…

…ary keys (r instead of r_prior etc.)
pymc-labs · Sep 15, 2023 · d865b99 · d865b99
1 parent 500251c
commit d865b99
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pymc_marketing/clv/models/beta_geo.py b/pymc_marketing/clv/models/beta_geo.py
@@ -59,10 +59,10 @@ class BetaGeoModel(CLVModel):
                 "recency":[30.73, 1.72, 0., 0., ...]
             }),
             model_config={
-                "r": pm.Gamma.dist(alpha=0.1, beta=0.1),
-                "alpha": pm.Gamma.dist(alpha=0.1, beta=0.1),
-                "a": pm.Gamma.dist(alpha=0.1, beta=0.1),
-                "b": pm.Gamma.dist(alpha=0.1, beta=0.1),
+                "r_prior": pm.Gamma.dist(alpha=0.1, beta=0.1),
+                "alpha_prior": pm.Gamma.dist(alpha=0.1, beta=0.1),
+                "a_prior": pm.Gamma.dist(alpha=0.1, beta=0.1),
+                "b_prior": pm.Gamma.dist(alpha=0.1, beta=0.1),
             },
             sampler_config={
                 "draws": 1000,