1. 程式人生 > >解決Keras下,imdb.load_data(num_words=10000)無法下載資料集的問題

解決Keras下,imdb.load_data(num_words=10000)無法下載資料集的問題

當我們按照deeplearning with python書裡面的程式碼教程來時,往往會出現資料集下載失敗的問題,例如執行下面一段程式碼

(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)

會去一個網站下載imdb.npz的資料集,這時很可能下載失敗,那麼怎麼辦呢?

可以在百度上先下載imdb.npz資料集,存放在一個資料夾內,然後程式碼改成如下:

(train_data, train_labels), (test_data, test_labels) = imdb.load_data(path="/home/cc/datasets/imdb.npz", num_words=10000)

哈哈,問題解決,是不是很簡單。