Keras 3 API 文档 / 内置小型数据集 / Fashion MNIST 数据集,MNIST 的替代品

Fashion MNIST 数据集,MNIST 的替代品

[源代码]

load_data 函数

keras.datasets.fashion_mnist.load_data()

加载 Fashion-MNIST 数据集。

这是一个包含 60,000 张 28x28 像素灰度图像的数据集,这些图像属于 10 个时尚类别,以及一个包含 10,000 张图像的测试集。 此数据集可以用作 MNIST 的直接替代品。

类别如下

标签 描述
0 T 恤/上衣
1 裤子
2 套头衫
3 连衣裙
4 外套
5 凉鞋
6 衬衫
7 运动鞋
8
9 踝靴

返回

NumPy 数组的元组:(x_train, y_train), (x_test, y_test)

x_trainuint8 NumPy 数组,包含形状为 (60000, 28, 28) 的灰度图像数据,表示训练数据。

y_trainuint8 NumPy 数组,包含标签(0-9 范围内的整数),形状为 (60000,),表示训练数据的标签。

x_testuint8 NumPy 数组,包含形状为 (10000, 28, 28) 的灰度图像数据,表示测试数据。

y_testuint8 NumPy 数组,包含标签(0-9 范围内的整数),形状为 (10000,),表示测试数据的标签。

示例

(x_train, y_train), (x_test, y_test) = fashion_mnist.load_data()
assert x_train.shape == (60000, 28, 28)
assert x_test.shape == (10000, 28, 28)
assert y_train.shape == (60000,)
assert y_test.shape == (10000,)

许可

Fashion-MNIST 的版权由 Zalando SE 持有。Fashion-MNIST 根据 MIT 许可证 授权。