由于默认是从amazon网站下载,被墙了,所以采取折中方案,使用浏览器下载好, 例如imdb.npz, 然后使用如下代码解析。
import numpy as np
# Load test data
imdb_data = np.load('imdb.npz')
imdb_data_dict = dict(zip(('train_data', 'train_label', 'test_data', 'test_label'), (imdb_data[k] for k in imdb_data)))
train_data, train_label, test_data, test_label = imdb_data_dict['train_data'], imdb_data_dict['train_label'], \
imdb_data_dict['test_data'], imdb_data_dict['test_label']