混淆矩阵(Confusion Matrix)是一种用来评估分类模型性能的可视化工具。它通过表格的形式展示实际类别与模型预测类别之间的对应关系,包括真正例(True Positive, TP)、假正例(False Positive, FP)、真负例(True Negative, TN)和假负例(False Negative, FN)。
混淆矩阵适用于二分类和多分类问题。在机器学习、深度学习、数据挖掘等领域中广泛应用,特别是在需要评估分类模型性能的场景中。
以下是使用scikit-learn
库绘制混淆矩阵的示例代码,假设我们有一组实际标签和预测标签:
import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
# 示例数据
y_true = np.array([0, 1, 0, 1, 0, 1, 0, 1])
y_pred = np.array([0, 0, 0, 1, 0, 1, 1, 1])
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 绘制混淆矩阵
disp = ConfusionMatrixDisplay(confusion_matrix=cm)
disp.plot(cmap=plt.cm.Blues)
plt.title('Confusion Matrix')
plt.show()
ValueError: Input contains NaN or infinity
错误怎么办?通过以上方法,可以在没有分类器的情况下绘制混淆矩阵,并解决常见的错误问题。
领取专属 10元无门槛券
手把手带您无忧上云