您可以使用以下代碼導入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)
變量中。此外,代碼還輸出了訓練集和測試集的形狀。