Skip to content

Instantly share code, notes, and snippets.

@rsepassi
Created February 25, 2019 21:52
Show Gist options
  • Save rsepassi/2fe0ce48956a4332e19e2fee299897bc to your computer and use it in GitHub Desktop.
Save rsepassi/2fe0ce48956a4332e19e2fee299897bc to your computer and use it in GitHub Desktop.
# Install: pip install tensorflow-datasets
import tensorflow_datasets as tfds
mnist_data = tfds.load("mnist")
mnist_train, mnist_test = mnist_data["train"], mnist_data["test"]
assert isinstance(mnist_train, tf.data.Dataset)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment