按照网上的教程加载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
求助各位!谢谢!
1个回答
我试了一下,好像是numpy版本的问题,numpy1.17的版本和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)
SofaSofa数据科学社区DS面试题库 DS面经
嗯,加上上面的代码就可以了,谢谢!
-
thatdeep
2019-08-29 01:34