TensorFlow提供了一個庫,可以直接用來自動下載與安裝MNIST。
MNIST里包含3個數據集:第一個是訓練數據集(mnist.train.images),另外兩個分別是測試數據集(mnist.test.images)和驗證數據集(mnist.validation)。
代碼中的one_hot=True,表示將樣本標簽轉化為one_hot編碼。
剛開始的打印信息是解壓數據集的意思。如果是第一次運行,還會顯示下載數據的相關消息。
接着打印出來的是訓練集的圖片信息,是一個55000行、784列的矩陣。即,訓練集里有55000張圖片。
1 from tensorflow.examples.tutorials.mnist import input_data 2 mnist = input_data.read_data_sets("MNIST_data/",one_hot=True) 3 print ('輸入數據:',mnist.train.images) 4 print ('輸入打印shape:',mnist.train.images.shape) 5 import pylab 6 im = mnist.train.images[1] 7 im = im.reshape(-1,28) 8 pylab.imshow(im) 9 pylab.show()
代碼的輸出結果如圖:

