import tensorflow as tf

        print("Test", i, "Prediction:", np.argmax(Ytr[nn_index]), \
            "True Class:", np.argmax(Yte[i]))
        # Calculate accuracy
        if np.argmax(Ytr[nn_index]) == np.argmax(Yte[i]):
            accuracy += 1. / len(Xte)
    print("Accuracy:", accuracy)

    # Make the log files in TensorBoard
    logs_path = "./logs"
    logWriter = tf.summary.FileWriter(logs_path, sess.graph)
