Update loss_functions.py

bfortuner · Jul 26, 2018 · 143b60a · 143b60a
1 parent a636f3a
commit 143b60a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/code/loss_functions.py b/code/loss_functions.py
@@ -10,10 +10,10 @@
 ### Functions ###
 
 def CrossEntropy(yHat, y):
-    if yHat == 1:
-      return -log(y)
+    if y == 1:
+      return -log(yHat)
     else:
-      return -log(1 - y)
+      return -log(1 - yHat)
 
 
 def Dice(yHat, y):