机器学习中的评价指标：F1-score

时间：2024-03-14 21:55:07

机器学习中的评价指标：F1-score

机器学习中的评价指标：F1-score

图：西雅图（Seattle）

Guderian出品

回顾：F1-score的计算过程

混淆矩阵（Confusion Matrix）

混淆矩阵也称误差矩阵，是表示精度评价的一种标准格式，用n行n列的矩阵形式来表示。

在机器学习领域，混淆矩阵用于衡量一个分类器的准确程度。对于二分类问题，将其样例根据真实类别和分类器的预测类别的组合划分为真正例(True Positive)\mathrm{(True \, Positive)}、假正例(False Positive)\mathrm{(False \, Positive)}、真反例(True Negative)\mathrm{(True \, Negative)}、假反例(False Negative)\mathrm{(False \, Negative)}四种情形。

对应混淆矩阵(Confusion Matrix)\mathrm{(Confusion \, Matrix)}如下表：

真实情况
预测结果
真
假
真
TP（真正例）
FN（假反例）
假
FP（假正例）
TN（真反例）

精确率（Precision）与召回率（Recall）

根据混淆矩阵，我们可以得到如下定义：

  • 精确率(Precision)\mathrm{(Precision)} P=TPTP+FP×100%\mathrm{P=\frac{TP}{TP+FP} \times 100\%}，表示真实情况和预测结果皆为真/预测结果为真
  • 召回率(Recall)\mathrm{(Recall)} R=TPTP+FN×100%\mathrm{R=\frac{TP}{TP+FN} \times 100 \%}，表示真实情况和预测结果皆为真/真实情况为真

F1分数（F1-score）

F1分数（F1−score\mathrm{F1-score}）即精确率与召回率的调和平均数（Harmonic Mean\mathrm{Harmonic \, Mean}），比算术平均数（Arithmetic Mean\mathrm{Arithmetic \, Mean}）的评价效果更好。

F1=21P+1R=2PRP+R\mathrm{F1=\frac{2}{\frac1{P}+\frac1{R}}=\frac{2PR}{P+R}}

直观理解为什么F1-score能评价模型优劣

在你训练的机器学习模型过程中，你往往希望能够兼顾精确率和召回率，并使用一个统一的单值评价指标来评价你的机器学习模型的训练效果。我们之所以使用调和平均而不是算术平均，是因为在算术平均中，任何一方对数值增长的贡献相当，任何一方对数值下降的责任也相当；而调和平均在增长的时候会偏袒较小值，也会惩罚精确率和召回率相差巨大的极端情况，很好地兼顾了精确率和召回率。

几何观点

机器学习中的评价指标：F1-score

上图为函数图象z=f(x,y)=2xyx+yz=f(x, y)=\frac{2xy}{x+y}。显而易见当x=1x=1，y=0y=0或者y=1y=1，x=0x=0时，z=0z=0，说明调和平均惩罚精确率和召回率相差很大的极端情况。

当x<yx<y时，梯度grad z\mathrm{grad} \, z在xx方向增长比yy方向快；当y<xy<x时，梯度grad z\mathrm{grad} \, z在yy方向增长比xx方向快，即调和平均在增长的时候会偏袒较小值。

代数观点

对函数f(x,y)=2xyx+yf(x, y)=\frac{2xy}{x+y}，其图象关于平面x=yx=y对称。分别求f(x,y)f(x,y)对xx和yy的偏导数：

​ ∂f(x,y)∂x=2y2(x+y)2\frac{\partial f(x,y)}{\partial x}=\frac{2y^2}{(x+y)^2}

​ ∂f(x,y)∂y=2x2(x+y)2\frac{\partial f(x,y)}{\partial y}=\frac{2x^2}{(x+y)^2}

当x<yx<y时，∂f(x,y)∂x>∂f(x,y)∂y\frac{\partial f(x,y)}{\partial x}>\frac{\partial f(x,y)}{\partial y}，梯度grad z\mathrm{grad} \, z在xx方向增长比yy方向快；

当y<xy<x时，∂f(x,y)∂y>∂f(x,y)∂x\frac{\partial f(x,y)}{\partial y}>\frac{\partial f(x,y)}{\partial x}，梯度grad z\mathrm{grad} \, z在yy方向增长比xx方向快，即调和平均在增长的时候会偏袒较小值。

可以看出，F1−score\mathrm{F1-score}是一种综合的评价指标,它要求总体高指标必须建立在同时满足高精确率和高召回率的情况之上。使得我们可以更方便快速地对不同模型进行比较，从而选择最优的那个模型。

Reference and lots of thanks：https://blog.csdn.net/otengyue/article/details/89426004

相关文章