加載MNIST數(shù)據(jù)
為了方便起見,我們已經(jīng)準(zhǔn)備了一個腳本來自動下載和導(dǎo)入MNIST數(shù)據(jù)集滓玖。它會自動創(chuàng)建一個 'MNIST_data' 的目錄來存儲數(shù)據(jù)行瑞。
import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
這里蔬啡,mnist是一個輕量級的類。它以Numpy數(shù)組的形式存儲著訓(xùn)練诵肛、校驗和測試數(shù)據(jù)集屹培。同時提供了一個函數(shù),用于在迭代中獲得minibatch,后面我們將會用到惫谤。