加载本地MNIST数据集(NPZ格式)

1、MNIST数据集下载( 提取码: MN4S)

下载

2、加载数据

import numpy as np

#加载数据
data= np.load('MNIST_data/mnist.npz', allow_pickle=True)
x_train, y_train = data['x_train'], data['y_train']
x_test, y_test = data['x_test'], data['y_test']

#打印形状
print(x_train.shape)
print(y_train.shape)
print(x_test.shape)
print(y_test.shape)

打印结果:

(60000, 28, 28)
(60000,)
(10000, 28, 28)
(10000,)

以上代码使用了NumPy库中的np.load函数,它可以从一个保存了NumPy数组的文件中加载数据。具体来说,np.load函数的第一个参数是文件名,allow_pickle=True表示可以加载序列化的数据。

你可能感兴趣的:(python,numpy,机器学习)