You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Python报错TypeError: unhashable type: 'numpy.ndarray'如何解决

报错原因

model.predict()返回的是所有类别的预测概率数组,你当前代码中的answer[0]是一个一维numpy数组(比如你做4分类任务的话,它的格式类似[0.02, 0.91, 0.05, 0.02]),而字典的key必须是整数、字符串这类可哈希的类型,numpy数组不支持作为key使用,因此触发该报错。

另外你代码中重复调用了两次model.predict(img),属于多余操作,第一次预测得到的answer已经包含所有需要的概率数据。

修复方案

只需要从预测概率数组中提取最大值对应的类别索引,用该整数索引查询字典即可,修改后的代码如下:

dic = test_dataset.class_indices
idc = {k:v for v, k in dic.items()}

img = load_img( r'C:\Users\sreep\Downloads\Alzheimer_s Dataset\test\NonDemented\26 (62).jpg', target_size = (224,224,3))
img = img_to_array(img)
img = img/255
imshow(img)
plt.axis('off')
img = np.expand_dims(img,axis=0)
answer = model.predict(img)
# 提取概率最大值对应的类别索引(整数类型,可作为字典key)
pred_class_idx = np.argmax(answer[0])
probability = round(np.max(answer)*100,2)

print(probability, '% chances are there that the image is',idc[pred_class_idx])

核心修改点说明:

  • 使用np.argmax(answer[0])拿到概率最高的类别对应的整数索引
  • 去掉重复的model.predict调用,直接复用第一次的预测结果提升效率

内容的提问来源于stack exchange,提问作者Sridara Pavan

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.09.24 06:36:04