From 3b7f58e81269167139de056b470dcd88b3701d35 Mon Sep 17 00:00:00 2001 From: ColtAllen Date: Fri, 14 Jul 2023 09:28:49 -0600 Subject: [PATCH] changed N to 4000 --- pymc_marketing/clv/models/pareto_nbd.py | 2 +- tests/clv/models/test_pareto_nbd.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pymc_marketing/clv/models/pareto_nbd.py b/pymc_marketing/clv/models/pareto_nbd.py index 01c31616..4c4474ff 100644 --- a/pymc_marketing/clv/models/pareto_nbd.py +++ b/pymc_marketing/clv/models/pareto_nbd.py @@ -633,7 +633,7 @@ def _population_distributions( # This is the shape if using fit_method="map" if self.fit_result.dims == {"chain": 1, "draw": 1}: - shape_kwargs = {"shape": 10000} + shape_kwargs = {"shape": 4000} else: shape_kwargs = {} diff --git a/tests/clv/models/test_pareto_nbd.py b/tests/clv/models/test_pareto_nbd.py index 7944e802..63b9fbef 100644 --- a/tests/clv/models/test_pareto_nbd.py +++ b/tests/clv/models/test_pareto_nbd.py @@ -361,7 +361,7 @@ def test_dropout_purchase_distributions(self) -> None: assert isinstance(customer_dropout, xarray.DataArray) assert isinstance(customer_purchase_rate, xarray.DataArray) - N = 10000 + N = 4000 lam = pm.Gamma.dist(self.r_true, self.alpha_true, size=N) mu = pm.Gamma.dist(self.s_true, self.beta_true, size=N) rtol = 0.05