MNIST
收藏数据集概述
TensorFlow Datasets 提供多种公共数据集,可通过 tf.data.Datasets 接口访问。
数据集列表
数据集使用示例
-
Python 代码示例: python import tensorflow_datasets as tfds import tensorflow as tf
查看可用数据集
print(tfds.list_builders())
加载数据集
ds_train = tfds.load(name="mnist", split="train", shuffle_files=True)
构建输入管道
ds_train = ds_train.shuffle(1000).batch(128).prefetch(10) for features in ds_train.take(1): image, label = features["image"], features["label"]
数据集构建器 (DatasetBuilder)
-
功能: 所有数据集通过
tfds.core.DatasetBuilder实现,提供tfds.builder和tfds.load两种访问方式。 -
示例: python import tensorflow_datasets as tfds
mnist_builder = tfds.builder(mnist) mnist_builder.download_and_prepare() ds = mnist_builder.as_dataset(split=train) info = mnist_builder.info
NumPy 使用
- 功能: 使用
tfds.as_numpy将tf.data.Dataset转换为 NumPy 数组。 - 示例: python train_ds = tfds.load("mnist", split="train") train_ds = train_ds.shuffle(1024).batch(128).repeat(5).prefetch(10) for example in tfds.as_numpy(train_ds): numpy_images, numpy_labels = example["image"], example["label"]
引用信息
-
引用格式:
@misc{TFDS, title = {{TensorFlow Datasets}, A collection of ready-to-use datasets}, howpublished = {url{https://www.tensorflow.org/datasets}}, }
数据集请求




