scale alpha by 1/scale so that var scales by scale factor

2026-09-10 12:37:53 +08:00 · 2020-03-27 11:34:12 +01:00
parent c52dfe9853
commit 47063478b7
1 changed files with 1 additions and 1 deletions
@@ -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)