Google I / O kehrt vom 18. bis 20. Mai zurück! Reservieren Sie Platz und erstellen Sie Ihren Zeitplan Registrieren Sie sich jetzt

TensorFlow-Datensätze: Eine Sammlung gebrauchsfertiger Datensätze.

TensorFlow Datasets ist eine Sammlung gebrauchsfertiger Datasets mit TensorFlow oder anderen Python ML-Frameworks wie Jax. Alle Datasets werden alstf.data.Datasets , wastf.data.Datasets und leistungsstarke Eingabe-Pipelines ermöglicht. Informationen zum Einstieg finden Sie im Handbuch und in unserer Liste der Datensätze .
import tensorflow.compat.v2 as tf
import tensorflow_datasets as tfds

# Construct a tf.data.Dataset
ds = tfds.load('mnist', split='train', shuffle_files=True)

# Build your input pipeline
ds = ds.shuffle(1024).batch(32).prefetch(tf.data.experimental.AUTOTUNE)
for example in ds.take(1):
  image, label = example["image"], example["label"]
In einem Notebook ausführen