improve numerical stability for kuma sample

xukai92 · Apr 28, 2019 · 3a52b96 · 3a52b96
1 parent 27a65a6
commit 3a52b96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distributions/beta.jl b/src/distributions/beta.jl
@@ -15,7 +15,7 @@ end
 function _u2logkumaraswamysample(T, u, kuma::BatchKumaraswamy)
     _one = one(T)
     _eps = eps(T)
-    return log.(_one .- exp.(log.(u) ./ kuma.b)) ./ kuma.a
+    return log.(_one .- exp.(log.(u .+ _eps) ./ kuma.b)) ./ kuma.a
 end
 
 """