Skip to content

Instantly share code, notes, and snippets.

@amirhfarzaneh

amirhfarzaneh/vgg13_mnist_loading_dataset.py

Created February 23, 2018 09:01
Show Gist options
  • Save amirhfarzaneh/692fa2cca157273e3c9d0de030271073 to your computer and use it in GitHub Desktop.
Save amirhfarzaneh/692fa2cca157273e3c9d0de030271073 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
vgg13_mnist_loading_dataset.py
# Loading the data (MNIST)
mnist = tf.contrib.learn.datasets.load_dataset("mnist")
train_data = mnist.train.images # Returns np.array
train_labels = np.asarray(mnist.train.labels, dtype=np.int32)
eval_data = mnist.test.images # Returns np.array
eval_labels = np.asarray(mnist.test.labels, dtype=np.int32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment