如何解决获取 MNIST 数据时出现 KeyError
我正在尝试使用 MNSIT 数据集,这是我一开始遇到的错误。 Error while fetching MNIST image
from sklearn.datasets import fetch_openml
mnist = fetch_openml('mnist_784')
x,y = mnist['data'],mnist['target']
digit = x[36001]
解决方法
x
是二维的;试试这个:
digit = x[36001,:]
既然您遇到了错误,这一定是由于 x 未被视为 numpy array
,请尝试以下两种方法:
digit = x.iloc[36001,:]
或
digit = x.to_numpy()[36001,:]
,
扩展我的评论,我认为 openml 的 MNIST 数据集最近(?)切换到返回一个 Pandas DataFrame 而不是一个 numpy 数组。另一个可能的修复是在函数调用中:
mnist = fetch_openml('mnist_784',as_frame=False)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。