多分类使用sklearn计算y_pred和y_prob

1.在开始计算时,给定类别为3类使用以下代码进行计算

python 复制代码
  y_pred = classifier.predict(X_test)

  y_pred_prob = classifier.predict_proba(X_test)
 

发现发现输出的y_pred标签没有对应于y_pred_prob中每一个最大值对应位置的标签,找不到存在的问题。

bash 复制代码
 print(y_test)
 print(y_pred)
 print(y_pred_prob)
-------------------------------------------

[2 2 2 ... 2 1 1]
[1 1 1 ... 0 1 1]
[[0.24505519 0.5998661  0.15507871]
 [0.03279259 0.68830022 0.27890719]
 [0.01111932 0.5753406  0.41354007]
 ...
 [0.82954333 0.02279303 0.14766364]
 [0.20383596 0.51148829 0.28467575]
 [0.03366171 0.92129916 0.04503913]]

多方面检查后,才发现

bash 复制代码
 print(pd.unique(y_pred))
#[1 2 0]

 print(pd.unique(y_test))
#[1 2 3]

----------------------------------------

原来是在使用sklearn计算y_pred时,类别标签是从0,开始,导致结果无法对应。

相关推荐
DisonTangor15 分钟前
阿里开源Qwen3-Omni-30B-A3B三剑客——Instruct、Thinking 和 Captioner
人工智能·语言模型·开源·aigc
独孤--蝴蝶17 分钟前
AI人工智能-机器学习-第一周(小白)
人工智能·机器学习
西柚小萌新18 分钟前
【深入浅出PyTorch】--上采样+下采样
人工智能·pytorch·python
丁学文武1 小时前
大语言模型(LLM)是“预制菜”? 从应用到底层原理,在到中央厨房的深度解析
人工智能·语言模型·自然语言处理·大语言模型·大模型应用·预制菜
fie88891 小时前
基于MATLAB的声呐图像特征提取与显示
开发语言·人工智能
文火冰糖的硅基工坊2 小时前
[嵌入式系统-100]:常见的IoT(物联网)开发板
人工智能·物联网·架构
刘晓倩2 小时前
实战任务二:用扣子空间通过任务提示词制作精美PPT
人工智能
shut up2 小时前
LangChain - 如何使用阿里云百炼平台的Qwen-plus模型构建一个桌面文件查询AI助手 - 超详细
人工智能·python·langchain·智能体
Hy行者勇哥2 小时前
公司全场景运营中 PPT 的类型、功能与作用详解
大数据·人工智能
FIN66683 小时前
昂瑞微:实现精准突破,攻坚射频“卡脖子”难题
前端·人工智能·安全·前端框架·信息与通信