Update lab-03-2-minimizing_cost_gradient_update.py (hunkim#70)

bkRyusim · hunkim · commit ee1340ae402f · 2017-03-17T21:54:07.000+09:00
Changed tf.reduce_sum to tf.reduce_mean. About issue hunkim#69
diff --git a/lab-03-2-minimizing_cost_gradient_update.py b/lab-03-2-minimizing_cost_gradient_update.py
@@ -17,7 +17,7 @@
 hypothesis = X * W
 
 # cost/loss function
-cost = tf.reduce_sum(tf.square(hypothesis - Y))
+cost = tf.reduce_mean(tf.square(hypothesis - Y))
 
 # Minimize: Gradient Descent using derivative: W -= learning_rate * derivative
 learning_rate = 0.1
