Back out multinomial heuristic adjustment at @adienes's request

JuliaStats · Apr 12, 2024 · df932f4 · df932f4
1 parent 1c3530c
commit df932f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/samplers/multinomial.jl b/src/samplers/multinomial.jl
@@ -53,7 +53,7 @@ function _rand!(rng::AbstractRNG, s::MultinomialSampler,
                 x::AbstractVector{<:Real})
     n = s.n
     k = length(s)
-    if k < n/50+2
+    if n^2 > k
         multinom_rand!(rng, n, s.prob, x)
     else
         # Use an alias table