多分类使用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,开始,导致结果无法对应。

相关推荐
Zero_to_zero123440 分钟前
NVSpeech_170k 数据集音频提取处理
人工智能·音视频
聚集的流星1 小时前
大模型提示词工程调优
人工智能
东方佑2 小时前
从音频到Token:构建原神角色语音识别模型的完整实践
人工智能·音视频·语音识别
dlraba8022 小时前
基于 OpenCV 与 SIFT 算法的指纹识别系统实现:从匹配到可视化
人工智能·opencv·计算机视觉
shizidushu2 小时前
Hugging Face NLP课程学习记录 - 3. 微调一个预训练模型
人工智能·学习·自然语言处理·微调·huggingface
格林威2 小时前
机器视觉在半导体制造中有哪些检测应用
人工智能·数码相机·yolo·计算机视觉·视觉检测·制造·相机
月岛雫-2 小时前
“单标签/多标签” vs “二分类/多分类”
人工智能·分类·数据挖掘
云卓SKYDROID3 小时前
无人机飞行速度模块技术要点概述
人工智能·无人机·飞行速度·高科技·云卓科技
币须赢3 小时前
英伟达Thor芯片套件9月发货 “物理AI”有哪些?
大数据·人工智能
格林威3 小时前
机器视觉检测如何使用360 度全景成像镜头进行AI 瑕疵检测
人工智能·深度学习·数码相机·机器学习·计算机视觉·视觉检测·相机