mnist数据集python导入
时间: 2023-07-09 22:38:26 浏览: 161
MNIST数据集转换为jpg格式Python代码 完整训练测试数据集
导入MNIST数据集需要使用Python中的第三方库,例如`tensorflow`或`keras`。以下是使用`keras`库导入MNIST数据集的示例代码:
```python
from keras.datasets import mnist
# 加载MNIST数据集
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
# 输出训练集和测试集的形状
print('train_images shape:', train_images.shape)
print('train_labels shape:', train_labels.shape)
print('test_images shape:', test_images.shape)
print('test_labels shape:', test_labels.shape)
```
这段代码会从Keras库中导入MNIST数据集,并将其分为训练集和测试集。训练集包含60000张手写数字图片,测试集包含10000张手写数字图片。输出的形状信息可以验证数据集的正确导入。
阅读全文