python怎么导入mnist数据集

lewis 8年前 (2017-06-27) 阅读数 10 #程序编程

文章标签 python

您可以使用以下代码导入MNIST数据集：

from tensorflow.keras.datasets import mnist

# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()

# 输出训练集和测试集的形状
print('训练集：', x_train.shape, y_train.shape)
print('测试集：', x_test.shape, y_test.shape)

这段代码使用了TensorFlow的keras.datasets模块中的mnist函数来加载MNIST数据集。加载后，数据集被拆分为训练集和测试集，分别存储在(x_train, y_train)和(x_test, y_test)变量中。此外，代码还输出了训练集和测试集的形状。