python混淆矩阵计算

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)