Skip to content

Instantly share code, notes, and snippets.

@rsepassi
Created February 25, 2019 21:54
Show Gist options
  • Select an option

  • Save rsepassi/a2a2011a15fedfceffc16f16629beea4 to your computer and use it in GitHub Desktop.

Select an option

Save rsepassi/a2a2011a15fedfceffc16f16629beea4 to your computer and use it in GitHub Desktop.
import tensorflow_datasets as tfds
datasets = tfds.load("mnist")
train_dataset, test_dataset = datasets["train"], datasets["test"]
assert isinstance(train_dataset, tf.data.Dataset)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment