欢迎访问宙启技术站
智能推送

使用run()函数在Python中下载和转换MNIST数据集的代码示例

发布时间:2023-12-22 22:21:53

要下载和转换MNIST数据集,可以使用TensorFlow提供的tf.keras.datasets模块。下面是一个使用run()函数下载和转换MNIST数据集的示例代码:

import tensorflow as tf
from tensorflow.keras.datasets import mnist

def download_and_convert_mnist():
    # 下载MNIST数据集
    (train_images, train_labels), (test_images, test_labels) = mnist.load_data()

    # 将图像数据归一化到0到1之间
    train_images = train_images / 255.0
    test_images = test_images / 255.0

    # 将图像数据转换为(float32)类型
    train_images = train_images.astype('float32')
    test_images = test_images.astype('float32')

    # 打印MNIST数据集的信息
    print("训练集图像形状:", train_images.shape)
    print("训练集标签形状:", train_labels.shape)
    print("测试集图像形状:", test_images.shape)
    print("测试集标签形状:", test_labels.shape)

    return train_images, train_labels, test_images, test_labels

if __name__ == '__main__':
    train_images, train_labels, test_images, test_labels = download_and_convert_mnist()

在这个示例中,首先使用mnist.load_data()函数下载MNIST数据集,并将数据分为训练集和测试集。

然后,对图像数据进行归一化处理,将像素值从0到255的整数转换为0到1之间的浮点数。

接下来,通过astype()函数将图像数据转换为float32类型,以便与后续的模型训练和预测操作兼容。

最后,打印出训练集和测试集的图像形状和标签形状,以验证数据加载和转换的正确性。

可以通过调用download_and_convert_mnist()函数来运行下载和转换MNIST数据集的代码,并获取处理后的训练集和测试集数据。