pickle.load() 时报错 raise AttributeError(attr + " not found")

最新推荐文章于 2025-06-06 13:40:32 发布

原创最新推荐文章于 2025-06-06 13:40:32 发布 · 3.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pickle

计算资源管理专栏收录该内容

55 篇文章

订阅专栏

本文解决了一个关于使用Python3的pickle.load()函数时遇到的错误。错误发生在尝试加载一个之前保存的数据集时，提示属性错误“shapenotfound”。解决办法是在pickle.dump()时添加protocol=0参数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题：我用python3 的pickle.load() 时报错。错误信息如下代码，
错误日志：

Traceback (most recent call last):
  File "./ncf.py", line 179, in main
    train_dataset = pickle.load(save_traind)
  File "/anaconda3-5.0.1/lib/python3.6/site-packages/scipy/sparse/dok.py", line 244, in __setitem__
    if (isintlike(i) and isintlike(j) and 0 <= i < self.shape[0]
  File "/anaconda3-5.0.1/lib/python3.6/site-packages/scipy/sparse/base.py", line 576, in __getattr__
    raise AttributeError(attr + " not found")
AttributeError: shape not found

找到了对策：原来是dump 的时候需要加参数 protocol=0
具体的：

abc_file = list(XXX)
with open('abc_asve.pkl','wb') as sf:
  pickle.dump(abc_file, sf, protocol=0)
with open('abc_save.pkl', 'rb') as rf:
  abc_read = pickler.load(sf)

Done!

参考：
https://github.com/joblib/joblib/issues/177