Add delta(1e-7) for cross_entropy_error

oreilly-japan · Dec 20, 2017 · ea91786 · ea91786
1 parent 899c2c4
commit ea91786
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/functions.py b/common/functions.py
@@ -53,7 +53,7 @@ def cross_entropy_error(y, t):
         t = t.argmax(axis=1)
 
     batch_size = y.shape[0]
-    return -np.sum(np.log(y[np.arange(batch_size), t])) / batch_size
+    return -np.sum(np.log(y[np.arange(batch_size), t] + 1e-7)) / batch_size
 
 
 def softmax_loss(X, t):