导入并处理IMDB数据集遇到的问题

还是采用本地下载的方法:
数据集网盘地址:链接:https://pan.baidu.com/s/1USzPMj413kK_Cg7E5fwstw
提取码:3hus

并添加到指定的文件夹下,比如我的地址是:C:\Users\DELL.keras\datasets

之后,在jupyter notebook(我用的是这个,当然可以其他)加载数据集

问题1:
导入并处理IMDB数据集遇到的问题_第1张图片解决方法


import numpy as np
# save np.load
np_load_old = np.load

# modify the default parameters of np.load
np.load = lambda *a,**k: np_load_old(*a, allow_pickle=True, **k)

# call load_data with allow_pickle implicitly set to true
(x_train, y_train), (x_test, y_test)= imdb.load_data(num_words= max_features)

# restore np.load for future normal usage
np.load = np_load_old

输入以上代码即可解决

问题2:
训练模型时,输入以下代码:

# 绘制训练 & 验证的准确率值
plt.plot(history.history['accuracy'])
plt.plot(history.history['val_accuracy'])
plt.title('Model accuracy')
plt.ylabel('Accuracy')
plt.xlabel('Epoch')
plt.legend(['Train', 'Test'], loc='upper left')
plt.show()

它会报错:KeyError:'accuracy’
解决办法:特别简单,把‘accuracy’改成‘acc’即可

你可能感兴趣的:(导入并处理IMDB数据集遇到的问题)