IMDB 集加载异常

2020-04-27  本文已影响0人  steamed_bun
问题说明:

使用如下语句加载imdb集的时候

(x_train, y_train), (x_test, y_test) = 
                                    imdb.load_data(path='imdb.npz', num_words=10000)

会报ValueError: Object arrays cannot be loaded when allow_pickle=False,意思就是要将allow_pickle设置为,true,但是imdb.load_data又不支持修改这个参数。网上的方法大部分是回退numpy的版本到1.16.2,但是不想回退¯\_(ツ)_/¯

开宗明义--解决办法:来自于点此

将上面的语句修改为:

# save np.load 
# 将原np.load先保存一下
np_load_old = np.load
# modify the default parameters of np.load
# 修改of np.load为 原np.load指定allow_pickle=True
np.load = lambda *a, **k: np_load_old(*a, allow_pickle=True, **k)
# 执行加载语句 此时语句内部使用的np.load就是指定allow_pickle=True的np.load了
(x_train, y_train), (x_test, y_test) = imdb.load_data(path='imdb.npz', num_words=10000)
# restore np.load for future normal usage
# 为了之后再次使用将np.load置回原np.load
np.load = np_load_old
上一篇下一篇

猜你喜欢

热点阅读