两行代码在keras中轻松导入mnist数据集

最新推荐文章于 2024-09-09 23:24:11 发布

我本无心沉溺～

最新推荐文章于 2024-09-09 23:24:11 发布

阅读量2.6k

点赞数 5

文章标签：深度学习机器学习人工智能

本文链接：https://blog.csdn.net/qq_53349891/article/details/121075092

版权

帮助大家避免走过的坑，话不多说直接开整

首先需要下载mnist数据集，网址MNIST handwritten digit database, Yann LeCun, Corinna Cortes and Chris Burgeshttp://yann.lecun.com/exdb/mnist/

下载好的文件名字如下

不用解压，放到下面路径即可，（要根据自己的keras安装路径选择文件放置位置）

放置好后输入下面代码即可

##########从keras文件夹中的datasets文件中导入mnist数据集
from keras.datasets import mnist


#########将数据集载入，X_train为训练集数据,Y_train为训练集标签，X_test为测试集数据,Y_test为测试集标签
(X_train,Y_train),(X_test,Y_test)=mnist.load_data()


#########打印查看数据
print('X_train.shape:'+str(X_train.shape))
print('Y_train.shape:'+str(Y_train.shape))
print('X_test.shape:'+str(X_test.shape))
print('Y_test.shape:'+str(Y_test.shape))

结果