在TensorFlow v1.9中下载MNIST数据集,可以通过以下步骤完成:
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
"MNIST_data/"
是指定数据集下载的目录,可以根据需要修改。read_data_sets
函数会自动下载MNIST数据集并将其分为训练集、验证集和测试集。one_hot=True
表示将标签转换为独热编码。train_images = mnist.train.images
train_labels = mnist.train.labels
validation_images = mnist.validation.images
validation_labels = mnist.validation.labels
test_images = mnist.test.images
test_labels = mnist.test.labels
注意:以上代码是基于TensorFlow v1.9版本的,如果使用其他版本,可能会有些许差异。此外,TensorFlow v2.x版本已经不再支持tensorflow.examples.tutorials.mnist
模块,而是使用tf.keras.datasets.mnist
模块进行MNIST数据集的下载。
领取专属 10元无门槛券
手把手带您无忧上云