From 86e6828091ca9efc08f630ebad93c01e9129cefe Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 7 May 2020 16:20:14 +0200 Subject: [PATCH] NB multiply 2nd moment by scale --- pts/modules/distribution_output.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pts/modules/distribution_output.py b/pts/modules/distribution_output.py index 64b9f42..c124e2c 100644 --- a/pts/modules/distribution_output.py +++ b/pts/modules/distribution_output.py @@ -136,9 +136,9 @@ class NegativeBinomialOutput(DistributionOutput): mu, alpha = distr_args if scale is not None: - # alpha += (scale - 1) / (scale * mu) # multiply 2nd moment by scale - alpha /= scale # multiply 2nd moment by sqrt(scale) mu *= scale + # alpha = alpha + (scale - 1) / (scale * mu) # multiply 2nd moment by scale + alpha += (scale - 1) / mu n = 1.0 / alpha p = mu * alpha / (1.0 + mu * alpha)