阅读量:4
您可以使用以下代码导入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)
变量中。此外,代码还输出了训练集和测试集的形状。