mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-06-27 19:32:05 +08:00
Revert "scale alpha so that the var stays the same"
This reverts commit 3c6bbf28d7.
This commit is contained in:
@@ -136,8 +136,8 @@ class NegativeBinomialOutput(DistributionOutput):
|
||||
mu, alpha = distr_args
|
||||
|
||||
if scale is not None:
|
||||
alpha = (1 - scale + mu*alpha)/(mu*scale*scale)
|
||||
mu = mu*scale
|
||||
mu *= scale
|
||||
alpha /= scale
|
||||
|
||||
n = 1.0 / alpha
|
||||
p = mu * alpha / (1.0 + mu * alpha)
|
||||
|
||||
Reference in New Issue
Block a user