Ensembles de données TensorFlow : une collection d'ensembles de données prêts à l'emploi.
TensorFlow Datasets est une collection d'ensembles de données prêts à l'emploi, avec TensorFlow ou d'autres frameworks Python ML, tels que Jax. Tous les jeux de données sont exposés en tant que
tf.data.Datasets
, permettant des pipelines d'entrée faciles à utiliser et hautes performances. Pour commencer, consultez le guide et notre liste de jeux de données .
import tensorflow 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.AUTOTUNE) for example in ds.take(1): image, label = example["image"], example["label"]Exécuter dans un ordinateur portable