From d3caa104ff3abc465bfa7949de7b994b19be84e4 Mon Sep 17 00:00:00 2001 From: Sam Abbott Date: Fri, 2 Aug 2024 09:18:35 +0100 Subject: [PATCH] get rid of all NB padding and clamping --- EpiAware/src/EpiObsModels/utils.jl | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/EpiAware/src/EpiObsModels/utils.jl b/EpiAware/src/EpiObsModels/utils.jl index 8fde14e1b..ccb0227f6 100644 --- 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