added accuracy to lab-07-2

jennykang · jennykang · commit 751661a5dd51 · 2017-02-23T20:54:10.000+08:00
diff --git a/lab-07-2-learning_rate_and_evaluation.py b/lab-07-2-learning_rate_and_evaluation.py
@@ -53,3 +53,10 @@
     plt.imshow(mnist.test.images[r:r + 1].reshape(28,
                                                   28), cmap='Greys', interpolation='nearest')
     plt.show()
+
+    # Test model
+    correct_prediction = tf.equal(tf.argmax(hypothesis, 1), tf.argmax(Y, 1))
+    # Calculate accuracy
+    accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))
+    print("Accuracy: ", accuracy.eval(session=sess, feed_dict={
+          X: mnist.test.images, Y: mnist.test.labels}))
