From 47063478b72ad142aa8301d33c741ec627d0c401 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Fri, 27 Mar 2020 11:34:12 +0100
Subject: [PATCH] scale alpha by 1/scale so that var scales by scale factor

---
 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 56e462c..05cb53c 100644
--- a/pts/modules/distribution_output.py
+++ b/pts/modules/distribution_output.py
@@ -139,7 +139,7 @@ class NegativeBinomialOutput(DistributionOutput):
 
         if scale is not None:
             mu *= scale
-            alpha /= torch.sqrt(scale)
+            alpha /= scale
 
         n = 1.0 / alpha
         p = mu * alpha / (1.0 + mu * alpha)