Update lab-09-2-xor-nn.py (hunkim#237)

* Update lab-09-2-xor-nn.py 1. Removed unnecessary codes. * Update lab-09-2-xor-nn.py Add numpy again. * Update lab-09-2-xor-nn.py Add numpy again. * Update lab-09-2-xor-nn.py Commit suggstion. Co-Authored-By: qoocrab <[email protected]> * Update lab-09-2-xor-nn.py Commit Suggestion. Co-Authored-By: qoocrab <[email protected]> * Update lab-09-2-xor-nn.py Commit Suggestion. Co-Authored-By: qoocrab <[email protected]> * Update lab-09-2-xor-nn.py Add f-string output. * Update lab-09-2-xor-nn.py
phutoan31299 · Jan 14, 2019 · c644f04 · c644f04
1 parent d1461d4
commit c644f04
Showing 1 changed file with 23 additions and 30 deletions.
diff --git a/lab-09-2-xor-nn.py b/lab-09-2-xor-nn.py
@@ -3,18 +3,9 @@
 import numpy as np
 
 tf.set_random_seed(777)  # for reproducibility
-learning_rate = 0.1
 
-x_data = [[0, 0],
-          [0, 1],
-          [1, 0],
-          [1, 1]]
-y_data = [[0],
-          [1],
-          [1],
-          [0]]
-x_data = np.array(x_data, dtype=np.float32)
-y_data = np.array(y_data, dtype=np.float32)
+x_data = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
+y_data = np.array([[0], [1], [1], [0]], dtype=np.float32)
 
 X = tf.placeholder(tf.float32, [None, 2])
 Y = tf.placeholder(tf.float32, [None, 1])
@@ -28,10 +19,8 @@
 hypothesis = tf.sigmoid(tf.matmul(layer1, W2) + b2)
 
 # cost/loss function
-cost = -tf.reduce_mean(Y * tf.log(hypothesis) + (1 - Y) *
-                       tf.log(1 - hypothesis))
-
-train = tf.train.GradientDescentOptimizer(learning_rate=learning_rate).minimize(cost)
+cost = -tf.reduce_mean(Y * tf.log(hypothesis) + (1 - Y) * tf.log(1 - hypothesis))
+train = tf.train.GradientDescentOptimizer(learning_rate=0.1).minimize(cost)
 
 # Accuracy computation
 # True if hypothesis>0.5 else False
@@ -44,25 +33,29 @@
     sess.run(tf.global_variables_initializer())
 
     for step in range(10001):
-        sess.run(train, feed_dict={X: x_data, Y: y_data})
+        _, cost_val = sess.run([train, cost], feed_dict={X: x_data, Y: y_data})
         if step % 100 == 0:
-            print(step, sess.run(cost, feed_dict={
-                  X: x_data, Y: y_data}), sess.run([W1, W2]))
+            print(step, cost_val)
 
     # Accuracy report
-    h, c, a = sess.run([hypothesis, predicted, accuracy],
-                       feed_dict={X: x_data, Y: y_data})
-    print("\nHypothesis: ", h, "\nCorrect: ", c, "\nAccuracy: ", a)
+    h, p, a = sess.run(
+        [hypothesis, predicted, accuracy], feed_dict={X: x_data, Y: y_data}
+    )
+
+    print(f"\nHypothesis:\n{h} \nPredicted:\n{p} \nAccuracy:\n{a}")
 
 
 '''
-Hypothesis:  [[ 0.01338218]
- [ 0.98166394]
- [ 0.98809403]
- [ 0.01135799]]
-Correct:  [[ 0.]
- [ 1.]
- [ 1.]
- [ 0.]]
-Accuracy:  1.0
+Hypothesis:
+[[0.01338216]
+ [0.98166394]
+ [0.98809403]
+ [0.01135799]] 
+Predicted:
+[[0.]
+ [1.]
+ [1.]
+ [0.]] 
+Accuracy:
+1.0
 '''