fixed multinomial_mode error on zero-probabilities

egtaonline · Feb 13, 2017 · 4a8f2ac · 4a8f2ac
1 parent 19ac4f9
commit 4a8f2ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gameanalysis/utils.py b/gameanalysis/utils.py
@@ -273,7 +273,7 @@ def multinomial_mode(p, n):
     notation follows: Gall 2003. Determination of the modes of a Multinomial
     distribution.
     """
-    f = p * (n + p.size/2)
+    f = (p + _TINY) * (n + p.size/2)
     k = f.astype(int)
     f -= k
     n0 = k.sum()