python 读取 pkl文件

2022-09-04 22:25:04 浏览数 (2)

使用python 的cPickle 库中的load函数,可以读取pkl文件的内容

代码语言:javascript复制
import cPickle as pickle 
fr = open('mnist.pkl') #open的参数是pkl文件的路径 
inf = pickle.load(fr) #读取pkl文件的内容 
fr.close() #关闭文件

上述代码有时会出现EOFError错误

此时将代码更改为一下形式即可解决(就是将前两行代码合成一行就行了)

代码语言:javascript复制
inf = pickle.load(open('mnist.pkl'))

以mnist为例,inf的类型为元祖tuple,他又包含了三个元组,分别对应训练集,验证集,测试集。每个元组中又包含两个numpy.ndarray,分别对应训练数据和label数据。训练数据的组成是由50000个含有784个元素的列表组成,每个列表代表一张图片。label数据集是由50000个元素组成的一维numpy.ndarray向量。

0 人点赞