Skip to content

Commit 7167d55

Browse files
committed
Tweak beta computation in the Gamma distribution
1 parent 709d9dc commit 7167d55

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

rand_distr/src/gamma.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -620,8 +620,12 @@ where
620620
if a > F::one() {
621621
// Algorithm BB
622622
let alpha = a + b;
623-
let beta =
624-
((alpha - F::from(2.).unwrap()) / (F::from(2.).unwrap() * a * b - alpha)).sqrt();
623+
624+
let two = F::from(2.).unwrap();
625+
let beta_numer = alpha - two;
626+
let beta_denom = two * a * b - alpha;
627+
let beta = (beta_numer / beta_denom).sqrt();
628+
625629
let gamma = a + F::one() / beta;
626630

627631
Ok(Beta {

0 commit comments

Comments
 (0)