get rid of all NB padding and clamping

CDCgov · Aug 2, 2024 · 7c8b12f · 7c8b12f
1 parent 7bdb361
commit 7c8b12f
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/EpiAware/src/EpiObsModels/utils.jl b/EpiAware/src/EpiObsModels/utils.jl
@@ -39,10 +39,9 @@ Compute the mean-cluster factor negative binomial distribution.
 A `NegativeBinomial` distribution object.
 """
 function NegativeBinomialMeanClust(μ, α)
-    _μ = clamp(μ, 1e-6, 1e17)
-    _α = clamp(α, 1e-6, Inf)
-    ex_σ² = (_α * _μ^2)
-    p = clamp(_μ / (_μ + ex_σ²), 1e-17, 1 - 1e-17)
-    r = clamp(_μ^2 / ex_σ², 1e-17, 1e17)
+    μ² = μ^2
+    ex_σ² = α * μ²
+    p = μ / (μ + ex_σ²)
+    r = μ² / ex_σ²
     return NegativeBinomial(r, p)
 end