Owl_base_stats_dist_gaussian.{std_gaussian_rvs,gaussian_rvs}: avoid i…

…nfinity on Random.float returning 0 #661 Signed-off-by: Edwin Török <[email protected]>
owlbarn · Mar 16, 2024 · fd8ec0f · fd8ec0f
1 parent 99ee1b3
commit fd8ec0f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/base/stats/owl_base_stats_dist_gaussian.ml b/src/base/stats/owl_base_stats_dist_gaussian.ml
@@ -20,8 +20,8 @@ let std_gaussian_rvs () =
     !_z1)
   else (
     _case := true;
-    _u1 := Random.float 1.;
-    _u2 := Random.float 1.;
+    _u1 := rand01_exclusive ();
+    _u2 := rand01_exclusive ();
     _z0 := sqrt (~-.2. *. log !_u1) *. cos (2. *. Owl_const.pi *. !_u2);
     _z1 := sqrt (~-.2. *. log !_u1) *. sin (2. *. Owl_const.pi *. !_u2);
     !_z0)
@@ -35,8 +35,8 @@ let gaussian_rvs ~mu ~sigma =
     mu +. (sigma *. !_z1))
   else (
     _case := true;
-    _u1 := Random.float 1.;
-    _u2 := Random.float 1.;
+    _u1 := rand01_exclusive ();
+    _u2 := rand01_exclusive ();
     _z0 := sqrt (~-.2. *. log !_u1) *. cos (2. *. Owl_const.pi *. !_u2);
     _z1 := sqrt (~-.2. *. log !_u1) *. sin (2. *. Owl_const.pi *. !_u2);
     mu +. (sigma *. !_z0))