tensorflow下载mnist数据集

视频

https://www.bilibili.com/video/BV1jK4y187yB?p=44

代码

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@Introduce :
@File      : demo_load_minist.py
@Time      : 2021-03-08 15:31
@Author    : xia hua dong
@Tel       : 173 179 76823
@Email     : 17317976823@163.com
@pip       : pip install 
"""
import tensorflow as tf
import matplotlib.pyplot as plt

# 确定要下载的数据集
mnist = tf.keras.datasets.mnist
# 下载数据集
# 下载路径默认为： C:\Users\xiahuadong\.keras\datasets\...
(train_x, train_y), (test_x, test_y) = mnist.load_data()
print("train_x的条数:", len(train_x))
print("train_y的条数:", len(train_y))
print("test_x的条数:", len(test_x))
print("test_y的条数:", len(test_y))
print("train_x的数据类型:", type(train_x))  # numpy数组
print("train_y的数据类型:", type(train_y))  # numpy数组
print("train_x的维度:", train_x.ndim)
print("train_x的行数、列数:", train_x.shape)
print("train_y的维度:", train_y.ndim)
print("train_y的行数、列数:", train_y.shape)
# 显示图片
plt.axis("off")
plt.imshow(train_x[0], cmap="gray")
plt.title(train_y[0])
plt.show()
# print("标签：", train_y[0])  # 打印标签