Updating MSE loss function

bfortuner · Nov 14, 2017 · 839d475 · 839d475
1 parent b504765
commit 839d475
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/loss_functions.py b/code/loss_functions.py
@@ -27,7 +27,7 @@ def MLE(yHat, y):
 
 
 def MSE(yHat, y):
-    return np.sum((yHat - y)**2) / 2.0
+    return np.sum((yHat - y)**2) / y.size
 
 def MSE_prime(yHat, y):
     return yHat - y