keras无法加载imdb文本:Object arrays cannot be loaded when allow_pickle=False

一世执手 提交于 2020-03-01 11:05:38

 

按照网上的教程加载imdb电影评论数据集

from keras.datasets import imdb
(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)

结果报错:ValueError: Object arrays cannot be loaded when allow_pickle=False

 numpy版本的问题,numpy的的版本和keras没有完全兼容

解决方法

#在前面加上这三行代码
import numpy as np
old = np.load
np.load = lambda *a,**k: old(*a,**k,allow_pickle=True)

#然后再运行原代码即可

from keras.datasets import imdb
(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!