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

相关推荐
AndrewHZ18 分钟前
【图像处理基石】如何入门大规模三维重建?
人工智能·深度学习·大模型·llm·三维重建·立体视觉·大规模三维重建
5G行业应用22 分钟前
【赠书福利,回馈公号读者】《智慧城市与智能网联汽车,融合创新发展之路》
人工智能·汽车·智慧城市
悟空胆好小31 分钟前
分音塔科技(BABEL Technology) 的公司背景、股权构成、产品类型及技术能力的全方位解读
网络·人工智能·科技·嵌入式硬件
探讨探讨AGV31 分钟前
以科技赋能未来,科聪持续支持青年创新实践 —— 第七届“科聪杯”浙江省大学生智能机器人创意竞赛圆满落幕
人工智能·科技·机器人
cwn_1 小时前
回归(多项式回归)
人工智能·机器学习·数据挖掘·回归
聚客AI1 小时前
🔥 大模型开发进阶:基于LangChain的异步流式响应与性能优化
人工智能·langchain·agent
CareyWYR2 小时前
每周AI论文速递(250707-250711)
人工智能
AI街潜水的八角2 小时前
深度学习图像分类数据集—五种电器识别分类
人工智能·深度学习·分类
众链网络2 小时前
AI进化论08:机器学习的崛起——数据和算法的“二人转”,AI“闷声发大财”
人工智能·算法·机器学习
生命是有光的2 小时前
【机器学习】机器学习基础
人工智能·机器学习