From 6bc281e5260ba7da1c9c737d5538ff7b27952a3a Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Sat, 21 Mar 2020 17:47:35 +0100 Subject: [PATCH] fix scaling of alpha --- pts/modules/distribution_output.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/modules/distribution_output.py b/pts/modules/distribution_output.py index 0377f2d..dff3eb9 100644 --- a/pts/modules/distribution_output.py +++ b/pts/modules/distribution_output.py @@ -124,7 +124,7 @@ class NegativeBinomialOutput(DistributionOutput): if scale is not None: mu *= scale - alpha *= torch.sqrt(scale + 1.0) + alpha /= torch.sqrt(scale) n = 1.0 / alpha p = mu * alpha / (1.0 + mu * alpha)