wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 19:32:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

NB multiply 2nd moment by scale

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-05-07 16:20:14 +02:00
parent 6d6630250c
commit 86e6828091
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/distribution_output.py
+2 -2
View File
@@ -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)