diff --git a/pts/modules/distribution_output.py b/pts/modules/distribution_output.py index ad08783..f1c9582 100644 --- a/pts/modules/distribution_output.py +++ b/pts/modules/distribution_output.py @@ -136,8 +136,8 @@ class NegativeBinomialOutput(DistributionOutput): mu, alpha = distr_args if scale is not None: - mu *= scale - alpha /= scale + alpha = (1 - scale + mu*alpha)/(mu*scale*scale) + mu = mu*scale n = 1.0 / alpha p = mu * alpha / (1.0 + mu * alpha)