登录
首页 >  文章 >  python教程

Python混淆矩阵函数详解与使用方法

时间:2025-10-21 17:46:36 222浏览 收藏

在Python机器学习中,`confusion_matrix()`函数是评估分类模型性能的利器。它通过`sklearn.metrics`库提供,用于生成混淆矩阵,直观展示模型在预测过程中的表现。该函数接收真实标签(`y_true`)和预测标签(`y_pred`)作为输入,输出一个矩阵,其中每个元素反映了模型将真实类别预测为特定类别的数量。对于二分类问题,混淆矩阵呈现为2x2的表格,包含真阳性(TP)、假阳性(FP)、真阴性(TN)和假阴性(FN)等关键指标。多分类问题则生成NxN矩阵,帮助分析模型在各个类别上的分类效果。通过解读混淆矩阵,开发者能够深入了解模型的优势与不足,为模型优化提供有力依据。

混淆矩阵是评估分类模型性能的工具,通过sklearn的confusion_matrix()函数比较真实与预测标签。对于二分类,它输出包含TP、FP、TN、FN的2x2矩阵;多分类则生成NxN矩阵,行表示真实类别,列表示预测类别,直观展示模型分类效果。

python confusion_matrix()是什么

在Python中,confusion_matrix() 是一个用于评估分类模型性能的函数,来自 scikit-learn 库(即 sklearn)。它通过比较真实标签和预测标签,生成一个矩阵,帮助我们直观地看出模型的分类效果。

什么是混淆矩阵?

混淆矩阵(Confusion Matrix)是一个表格,用来描述分类模型在测试数据上的表现。对于二分类问题,它包含四个关键结果:

  • TP(True Positive):实际为正类,预测也为正类
  • FP(False Positive):实际为负类,预测为正类
  • TN(True Negative):实际为负类,预测也为负类
  • FN(False Negative):实际为正类,预测为负类

如何使用 confusion_matrix()?

你可以通过以下方式调用这个函数:

from sklearn.metrics import confusion_matrix

示例数据

y_true = [1, 0, 1, 1, 0, 1] # 真实标签 y_pred = [1, 0, 0, 1, 0, 1] # 模型预测标签

生成混淆矩阵

cm = confusion_matrix(y_true, y_pred) print(cm)

输出结果是一个二维数组:

[[2 0] [1 3]]

其中:

  • cm[0][0]:TN(真实0,预测0)
  • cm[0][1]:FP(真实0,预测1)
  • cm[1][0]:FN(真实1,预测0)
  • cm[1][1]:TP(真实1,预测1)

多分类情况下的使用

confusion_matrix() 同样适用于多分类任务。例如有三个类别(0, 1, 2),输出将是一个 3x3 的矩阵,每一行代表真实类别,每一列代表预测类别。

基本上就这些。这个函数不复杂但非常实用,是分析分类模型错误类型的基础工具。

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《Python混淆矩阵函数详解与使用方法》文章吧,也可关注golang学习网公众号了解相关技术文章。

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>