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

revert back change and added comments

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-04-07 11:05:17 +02:00
parent 5b57396cea
commit 2d8f6d31f0
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/distribution_output.py
+2 -1
View File
@@ -136,7 +136,8 @@ class NegativeBinomialOutput(DistributionOutput):
mu, alpha = distr_args
if scale is not None:
alpha = alpha + (scale - 1) / (scale * mu)
# alpha += (scale - 1) / (scale * mu) # multiply 2nd moment by scale
alpha /= scale # multiply 2nd moment by sqrt(scale)
mu *= scale
n = 1.0 / alpha