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")