sincerely fixed softmax

cipri-tom · Jan 27, 2017 · 3d5f73c · 3d5f73c
1 parent 760ab51
commit 3d5f73c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/starter.py b/scripts/starter.py
@@ -14,7 +14,8 @@
 
 def softmax(x, tau):
     """ Returns softmax probabilities with temperature tau"""
-    e_x = np.exp(x / tau)
+    x = x / tau
+    e_x = np.exp(x - np.max(x))
     return e_x / e_x.sum()
 
 class Agent():