1. 程式人生 > >Python 讀取 pkl檔案

Python 讀取 pkl檔案

 

使用python 的cPickle 庫中的load函式,可以讀取pkl檔案的內容

import cPickle as pickle

fr = open('mnist.pkl')    #open的引數是pkl檔案的路徑
inf = pickle.load(fr)       #讀取pkl檔案的內容
fr.close()                       #關閉檔案
 

上述程式碼有時會出現EOFError錯誤

此時將程式碼更改為一下形式即可解決(就是將前兩行程式碼合成一行就行了)

inf = pickle.load(open('mnist.pkl'))

以mnist為例,inf的型別為元祖tuple,他又包含了三個元祖,分別對應訓練集,驗證集,測試集。每個元祖中又包含兩個numpy.ndarray,分別對應訓練資料和label資料。訓練資料的組成是由50000個含有784個元素的列表組成,每個列表代表一張圖片。label資料集是由50000個元素組成的一維numpy.ndarray向量。