python混淆矩阵计算

最新推荐文章于 2024-05-17 14:08:45 发布

cxxx17

最新推荐文章于 2024-05-17 14:08:45 发布

阅读量545

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/weixin_42262721/article/details/120656865

版权

python 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

from sklearn.metrics import confusion_matrix
y_true = ["cat", "ant", "cat", "cat", "ant", "bird"]
y_pred = ["ant", "ant", "cat", "cat", "ant", "cat"]
cm = confusion_matrix(y_true, y_pred, labels=["ant", "bird", "cat"])
print(cm.shape) # (3,3)

计算多分类的recall，precision，f1得分

from sklearn.metrics import multilabel_confusion_matrix
mcm = multilabel_confusion_matrix(y_true, y_pred, labels=["ant", "bird", "cat"])
print(mcm.shape) # (3, 2, 2)