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

  统计/机器学习 自然语言处理 Python TensorFlow    浏览次数:5042        分享
0

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

求助各位!谢谢!

 

thatdeep   2019-08-28 06:51



   1个回答 
2

我试了一下,好像是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面经

strong.man   2019-08-28 09:16

嗯,加上上面的代码就可以了,谢谢! - thatdeep   2019-08-29 01:34


  相关讨论

keras加载imdb数据时num_words参数是干嘛的?

Keras模型针对同一数据集做fit和evaluation,accuracy值却不相同

keras和tensorflow有什么关系?

keras load_model报错

keras sequential model怎么增加一个dropout层?

为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?

训练完keras模型后怎么画出loss和val_loss的对比图?

怎么在keras中加载MNSIT数据集?

keras里的predict,predict_classes,predict_proba有什么区别?

keras.models.Sequential()是一种什么模型

  随便看看

怎么把dataframe的一列转成整数类型

PCA降维之前为什么要先标准化?

wide-deep模型里的wide和deep分别是指什么意思?

xgboost的gblinear是什么意思?

numpy.array转换为图片并显示出来