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

 

按照网上的教程加载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)

 

你可能感兴趣的:(人工智能)