def main(unused_argv):
tf.logging.set_verbosity(tf.logging.DEBUG)
mnist = Mnist("model")
# download training and eval data
mnist_data = tf.contrib.learn.datasets.load_dataset("mnist")
train_data = mnist_data.train.images # Returns np.array
train_labels = np.asarray(mnist_data.train.labels, dtype=np.int32)
eval_data = mnist_data.test.images # Returns np.array
eval_labels = np.asarray(mnist_data.test.labels, dtype=np.int32)
mnist.train(train_data, train_labels)
results = mnist.evaluate(eval_data, eval_labels)
print(results)
from mnist import Mnist
mnist = Mnist()
mnist.train(20000)
mnist.save("model.ckpt")
mnist.close()
print("done")