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

相关推荐
掘金安东尼22 分钟前
OCR的新高度?PaddleOCR-VL 与 DeepSeek-OCR 的技术与应用横评
人工智能
Aurora-silas30 分钟前
LLM微调尝试——MAC版
人工智能·pytorch·深度学习·macos·机器学习·语言模型·自然语言处理
空白到白41 分钟前
BERT,GPT,ELMO模型对比
人工智能·gpt·自然语言处理·bert
空白到白42 分钟前
BERT-微调任务
人工智能·自然语言处理·bert
文火冰糖的硅基工坊1 小时前
[人工智能-大模型-20]:对比 Copilot 与国产替代方案(如通义灵码、百度Comate)
人工智能·百度·copilot
岛屿旅人1 小时前
智库报告美国如何打网络战
网络·人工智能·安全·web安全
XIAO·宝2 小时前
深度学习------YOLOV3
人工智能·深度学习·yolo
AI科技星2 小时前
垂直原理:宇宙的沉默法则与万物运动的终极源头
android·服务器·数据结构·数据库·人工智能
开始学AI2 小时前
ChatClimate:让对话式人工智能立足于气候科学
人工智能
学习路上_write2 小时前
神经网络初次学习收获
人工智能·python