查准率、查全率、F1、混淆矩阵

from sklearn.metrics import precision_score
from sklearn.metrics import recall_score
from sklearn.metrics import f1_score
from sklearn.metrics import confusion_matrix

y_true = [0, 1, 1, 0, 1, 1, 0, 1, 1, 1, 0, 0 ,1]
y_pred = [0, 1, 1, 0, 1, 1, 0, 1, 1, 1, 0, 1 ,1]

precision_score(y_true = y_true, y_pred = y_pred)  # 0.88888888888888884
recall_score(y_true = y_true, y_pred = y_pred)     # 1.0
f1_score(y_true = y_true, y_pred = y_pred)         # 0.94117647058823528
confusion_matrix(y_true = y_true, y_pred = y_pred)
# array([[4, 1],
#        [0, 8]])
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容