混淆矩阵的生成(python实现,含机器学习方法)
from sklearn.metrics import confusion_matrix
# 假设以下是真实的目标值和预测值
target = ["cat", "dog", "cat", "cat", "dog"]
prediction = ["cat", "dog", "cat", "dog", "cat"]
# 生成混淆矩阵
cm = confusion_matrix(target, prediction)
# 打印混淆矩阵
print(cm)
这段代码使用了sklearn.metrics
模块中的confusion_matrix
函数来生成混淆矩阵。target
和prediction
分别是真实的类别标签和模型的预测结果。生成的混淆矩阵会打印出来。这是一个很直接的例子,展示了如何使用现有的库来快速生成混淆矩阵。
评论已关闭