当前位置:   article > 正文

sklearn.metrics 用法详解

sklearn.metrics

1. 用法概览

1.1 分类
函数功能
metrics.accuracy_score准确率
metrics.balanced_accuracy_score在类别不均衡的数据集中,计算加权准确率
metrics.top_k_accuracy_score获得可能性最高的k个类别
metrics.average_precision_score根据预测分数计算平均精度 (AP)
metrics.brier_score_lossBrier 分数损失
metrics.f1_scoreF1 score
metrics.log_loss交叉熵损失
metrics.precision_score精确率
metrics.recall_score召回率
metrics.jaccard_scoreJaccard 相似系数得分
metrics.roc_auc_score根据预测分数计算 Area Under the Receiver Operating Characteristic Curve(ROC AUC) 下的面积
metrics.cohen_kappa_score衡量注释间一致性的统计量
1.2 聚类
函数功能
metrics.adjusted_mutual_info_score两个聚类之间的调整互信息(AMI)
metrics.adjusted_rand_score调整兰德指数
metrics.completeness_score给定GT的集群标记的完整性度量
metrics.fowlkes_mallows_score测量一组点的两个聚类的相似性
metrics.homogeneity_score同质性指标
metrics.mutual_info_score互信息
metrics.normalized_mutual_info_score标准化互信息
metrics.rand_score兰德指数
metrics.v_measure_scoreV测度得分
1.3 回归
函数功能
metrics.explained_variance_score解释方差回归评分函数
metrics.mean_absolute_error平均绝对误差
metrics.mean_squared_error均方误差
metrics.mean_squared_log_error平均平方对数误差
metrics.median_absolute_error中位数绝对误差
metrics.r2_score R 2 R^2 R2(确定系数)

2. 数学原理

主要记录一下关于分类部分的数学原理。准确率 - accuracy,精确率 - precision,召回率 - recall,
F1值 - F1-score,ROC曲线下面积 - ROC-AUC (area under curve),PR曲线下面积 - PR-AUC。

对于一个二分类问题,假设真实标签y_labels=[1,1,0,1,1,0,0,0],我们预测的结果y_scores=[0.8,0.9,0.6,0.3,0.7,0.1,0.1,0.6]。假设threshold=0.5。那么可以得到y_preds=[1,1,1,0,1,0,0,1]。这时我们可以得到混淆矩阵(confusion matrix)为:

图1:混淆矩阵

图1:混淆矩阵

混淆矩阵中所对应的每一个值的含义如下:
图2:混淆矩阵的含义

图2:混淆矩阵的含义

那么:
准确率= T P + T N T P + T N + F P + F N \frac{TP+TN}{TP+TN+FP+FN} TP+TN+FP+FNTP+TN,精准率= T P T P + F P \frac{TP}{TP+FP} TP+FPTP,召回率= T P T P + F N \frac{TP}{TP+FN} TP+FNTP,F1-scores= 2 ∗ P r e c i s i o n ∗ R e c a l l P r e c i s i o n + R e c a l l \frac{2*Precision*Recall}{Precision+Recall} Precision+Recall2PrecisionRecall

ROC/AUC的概念

ROC(Receiver Operating Characteristic)曲线,又称接受者操作特征曲线。该曲线最早应用于雷达信号检测领域,用于区分信号与噪声。后来人们将其用于评价模型的预测能力,ROC曲线是基于混淆矩阵得出的。

灵敏度(Sensitivity)= T P T P + F N \frac{TP}{TP+FN} TP+FNTP,特异度(Specificity)= T N F P + T N \frac{TN}{FP+TN} FP+TNTN
真正率(TPR)= 灵敏度= T P T P + F N \frac{TP}{TP+FN} TP+FNTP,假正率(FPR) = 1- 特异度 = F P F P + T N \frac{FP}{FP+TN} FP+TNFP

在上述二分类的例子中,我们取threshold=0.5可以的到一个y_predsthreshold从0取到1就可以得到不同的y_preds,进而计算出不同的(FPR,TPR)对。它们在坐标轴上对应了一条曲线,这条曲线就是ROC曲线,曲线下的面积就是AUC的值。如下图:
图3:ROC 曲线

图3:ROC 曲线

多分类的计算
metrics.cohen_kappa_score:继续等待填坑

3. 实例

以之前的数据来计算每一个度量指标的值,这里用metrics.classification_report
metrics.classification_report(y_true, y_pred, *, labels=None, target_names=None, sample_weight=None, digits=2, output_dict=False, zero_division='warn')
注意到这里的参数是y_pred而不是y_score,所以它只能计算F1-score,而不能计算AUC值。
返回值的格式如下:
{'label 1': {'precision':0.5, 'recall':1.0, 'f1-score':0.67, 'support':1}, 'label 2': { ... }, ... }

from sklearn import metrics
import matplotlib.pyplot as plt

y_labels = [1,1,0,1,1,0,0,0]
y_scores=[0.8,0.9,0.6,0.3,0.2,0.1,0.1,0.6]
y_preds = [1,1,1,0,1,0,0,1] # threshold=0.5
report = metrics.classification_report(y_labels,y_preds)
fpr, tpr, thresholds = metrics.roc_curve(y_labels,y_scores)
auc = metrics.auc(fpr,tpr)
plt.plot(fpr,tpr,'*-')
plt.ylabel('TPR')
plt.xlabel('FPR')
plt.title('ROC curve')
print(report)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14

得到ROC曲线如图3所示,report的值如下:
图4:report结果

图4:report结果

参考链接:
[1] https://scikit-learn.org/stable/modules/model_evaluation.html
[2] https://blog.csdn.net/qq_27575895/article/details/83781069
[3] https://laurenoakdenrayner.com/2018/01/07/the-philosophical-argument-for-using-roc-curves/

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/142822
推荐阅读
相关标签
  

闽ICP备14008679号