下載並使用MNIST數據集


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()


代碼的輸出結果如圖:


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM