一、概述
分类模型是机器学习中一种最常见的问题模型,在许多问题场景中有着广泛的运用,是模式识别问题中一种主要的实现手段。分类问题概况起来就是,对一堆高度抽象了的样本,由经验标定了每个样本所属的实际类别,由特定算法训练得到一个分类器,输入样本属性即自动计算出其所属类别,从而完成特定的识别任务。依实现原理的不同,分类算法有很多种,常见的如支持向量机、决策树、k近邻、朴素贝叶斯…,以及设计了的各种人工神经网络等。在特定问题中,算法性能的评价是一个重要的方面,它一般由学习器在测试数据中的性能表现来直接衡定,常见的评价指标有准确率、精准度(查准率)、召回率(查全率)、F1值、ROC-AUC值、Kappa系数等。
二、评价指标——二分类情形
二分类模型是较常见和较一般化的情形,它预测的混淆矩阵可表示为
其中,
TP(True Positive,真正例)为被正确划分为正例的样本数;
FP(False Positive,假正例)为被错误划分为正例的样本数;
TN(True Negative,真负例)为被正确划分为负例的样本数;
FN(False Negative,假负例)为被错误划分为负例的样本数。
1.准确率(accuracy)
准确率是被预测正确的样本数与样本总数
N
N
N的比值,比较直观地反映了学习器的性能。表达式为
A
c
c
u
r
a
c
y
=
T
P
+
T
N
N
Accuracy=\frac{TP+TN}{N}
Accuracy=NTP+TN
2. 精准度(precision)
精准度是TP(真正例)占预测为正例样本数的百分比,它表征了预测的正例中有多少有效成分,也叫做查准率。表达式为
p
r
e
c
i
s
i
o
n
=
T
P
T
P
+
F
P
precision=\frac{TP}{TP+FP}
precision=TP+FPTP
3. 召回率(recall)
召回率是TP(真正例)与真实正样本数的比值,它表征了所有的有效内容有多少被检测了出来,也叫做查全率、灵敏度(sensitive)。表达式为
r
e
c
a
l
l
=
T
P
T
P
+
F
N
recall=\frac{TP}{TP+FN}
recall=TP+FNTP
4. F度量(F-Measure, F-Score)
我们希望学习器的查准率和查全率都高,而查准率和查全率往往是一对矛盾的度量,查准率高时,查全率往往偏低;查全率高时,查准率往往偏低。此时可以用F度量来综合评估精准度(查准率)和召回率(查全率),数值越高学习器越理想。它的一般形式记作
F
β
F_\beta
Fβ,表达式为
F
β
=
(
1
+
β
2
)
⋅
p
r
e
c
i
s
i
o
n
⋅
r
e
c
a
l
l
β
2
⋅
p
r
e
c
i
s
i
o
n
+
r
e
c
a
l
l
F_\beta=\frac{(1+\beta^2)\cdot precision\cdot recall}{\beta^2\cdot precision + recall}
Fβ=β2⋅precision+recall(1+β2)⋅precision⋅recall
式中的
β
\beta
β 值能够表达出对查准率/查全率的不同偏好,如F1分数认为召回率和精准度同等重要,F2分数认为召回率的重要程度是精准度的两倍,而F0.5分数认为召回率的重要程度是精准度的一半。其中,F1度量是最常用的一种,表达式为
F
1
=
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
F_1=\frac{2\cdot precision\cdot recall}{precision+recall}
F1=precision+recall2⋅precision⋅recall
注:
F1由调和平均进行定义: 1 F 1 = 1 2 ( 1 p r e c i s i o n + 1 r e c a l l ) \frac{1}{F_1}=\frac{1}{2}\left( \frac{1}{precision}+\frac{1}{recall} \right) F11=21(precision1+recall1) F β F_\beta Fβ由加权调和平均进行定义:
1 F β = 1 1 + β 2 ( 1 p r e c i s i o n + β 2 r e c a l l ) \frac{1}{F_\beta}=\frac{1}{1+\beta^2 }\left( \frac{1}{precision}+\frac{\beta^2}{recall} \right) Fβ1=1+β21(precision1+recallβ2)
5.ROC曲线与AUC值
ROC(Receiver Operating Characteristic,受试者工作特征)曲线是以真正率(TPR)为纵轴、假正率(FPR)为横轴绘制的曲线,AUC(Area Under ROC Curve)是ROC曲线下的面积,它们能够在一定程度上衡量分类器的性能,示意图如图所示。
TPR(true positive rate,真正率)、FPR(false positive rate,假正率)表达式为
T
P
R
=
T
P
T
P
+
F
N
TPR=\frac{TP}{TP+FN}
TPR=TP+FNTP
F
P
R
=
F
P
F
P
+
T
N
FPR=\frac{FP}{FP+TN}
FPR=FP+TNFP
ROC与AUC的度量,对二分类问题来说是更直观的,它直接地关注于正例样本,通过正例样本的整体表现来衡量学习器的性能。而这个正例类的指定,则由实际问题情形而确定。
ROC曲线的绘制过程为,对测试样本集中的样本,将待考察的类别规定为正例类,学习器计算得到每个样本归属于正例类的概率,对所有样本依概率进行排序,起始时将划分阈值设为最大,即将所有样本划为负样本,此时TPR和FPR均为0,坐标轴上在(0,0)处标定了一个点。然后调整划分阈值,依次将每个样本划为正样本,TPR、FPR值随之增大,连接过程中标定的坐标点即得到ROC曲线。易知,若一个学习器的ROC曲线被另一个学习器的曲线完全包住,则后者的性能优于前者;若两个学习器的曲线发生交叉,则可比较ROC曲线下的面积,即AUC(Area Under ROC Curve),面积越大,性能越好。
假定ROC曲线由坐标点{(x1,y1),(x2,y2),…,(xn,yn)}按序连接而成,则AUC值估算的表达式为
A
U
C
=
1
2
∑
i
=
1
n
−
1
(
x
i
+
1
−
x
i
)
⋅
(
y
i
+
y
i
+
1
)
AUC=\frac{1}{2}\sum_{i=1}^{n-1}{\left( x_{i+1}-x_i \right)\cdot\left( y_i+y_{i+1} \right)}
AUC=21i=1∑n−1(xi+1−xi)⋅(yi+yi+1)
6.Kappa系数
Kappa系数是统计分析中一种进行一致性检验的常见方法,能够较大程度上反映出分类算法的性能表现。它的计算同样基于混淆矩阵,反映的是预测结果与实际结果的一致性,其取值范围为[-1,1],值越大说明一致性越高,性能越好;反之性能越差。假设样本总数为N,Kappa系数定义为
k
a
p
p
a
=
p
o
−
p
e
1
−
p
e
kappa=\frac{p_o-p_e}{1-p_e}
kappa=1−pepo−pe
其中,
p
o
=
1
N
(
T
P
+
T
N
)
p_o=\frac{1}{N}\left( TP+TN \right)
po=N1(TP+TN)是整体准确率,
p
e
=
1
N
2
[
(
T
P
+
F
N
)
⋅
(
T
P
+
F
P
)
+
(
F
P
+
T
N
)
⋅
(
F
N
+
T
N
)
]
p_e=\frac{1}{N^2}\left[ \left( TP+FN \right)\cdot \left( TP+FP \right)+\left( FP+TN \right)\cdot \left( FN+TN \right) \right]
pe=N21[(TP+FN)⋅(TP+FP)+(FP+TN)⋅(FN+TN)]表偶然一致性。
三、评价指标——多分类情形
多分类情形下的混淆矩阵表示为
多分类问题往往能够转化为多个二分类问题,对它们性能的评价可通过宏计算或微计算的方式进行。
1.准确率(accuracy)
准确率即混淆矩阵主对角线上的值之和与样本总数N的比值
A
c
c
u
r
a
c
y
=
1
N
∑
i
=
1
K
x
i
i
Accuracy=\frac{1}{N}\sum_{i=1}^{K}{x_{ii}}
Accuracy=N1i=1∑Kxii
2.精准度、召回率、F1值
对于多分类任务,各个类别两两之间即对应了一个混淆矩阵,计算查准率、查全率、F1值的方式有两种,一种是对各混淆矩阵计算得到的指标值对应求平均,得到宏查准率(macro-P)、宏查全率(macro-R)和宏F1值(macro-F1);另一种是将各混淆矩阵对应位置处的值求平均,记为
T
P
ˉ
\bar{TP}
TPˉ、
F
P
ˉ
\bar{FP}
FPˉ、
T
N
ˉ
\bar{TN}
TNˉ、
F
N
ˉ
\bar{FN}
FNˉ,由这些平均的值计算得到微查准率(micro-P)、微查全率(micro-R)和微F1值(micro-F1)。假设共有K个类别,那么混淆矩阵数量为
C
K
2
=
K
⋅
(
K
−
1
)
2
C_{K}^{2}=\frac{K\cdot(K-1)}{2}
CK2=2K⋅(K−1),记为n,则
宏计算的指标值表示为
m
a
c
r
o
−
P
=
1
n
∑
i
=
1
n
p
r
e
c
i
s
i
o
n
i
macro_{-}P=\frac{1}{n}\sum_{i=1}^{n}{precision_i}
macro−P=n1i=1∑nprecisioni
m
a
c
r
o
−
R
=
1
n
∑
i
=
1
n
r
e
c
a
l
l
i
macro_{-}R=\frac{1}{n}\sum_{i=1}^{n}{recall_i}
macro−R=n1i=1∑nrecalli
m
a
c
r
o
x
−
F
1
=
2
∗
m
a
c
r
o
−
P
∗
m
a
c
r
o
−
R
m
a
c
r
o
−
P
+
m
a
c
r
o
−
R
macrox_{-}F1=\frac{2\ast macro_{-}P\ast macro_{-}R}{macro_{-}P+macro_{-}R}
macrox−F1=macro−P+macro−R2∗macro−P∗macro−R
微计算的指标值表示为
m
i
c
r
o
−
P
=
T
P
ˉ
T
P
ˉ
+
F
P
ˉ
micro_{-}P=\frac{\bar{TP}}{\bar{TP}+\bar{FP}}
micro−P=TPˉ+FPˉTPˉ
m
i
c
r
o
−
R
=
T
P
ˉ
T
P
ˉ
+
F
N
ˉ
micro_{-}R=\frac{\bar{TP}}{\bar{TP}+\bar{FN}}
micro−R=TPˉ+FNˉTPˉ
m
i
c
r
o
−
F
1
=
2
∗
m
i
c
r
o
−
P
∗
m
i
c
r
o
−
R
m
i
c
r
o
−
P
+
m
i
c
r
o
−
R
micro_{-}F1=\frac{2\ast micro_{-}P \ast micro_{-}R}{micro_{-}P+micro_{-}R}
micro−F1=micro−P+micro−R2∗micro−P∗micro−R
3. ROC曲线与AUC值
多分类情形下的ROC与AUC同样可以以“宏”或“微”的形式进行计算。
方式一:
对于K(K>2)类分类问题,分别将每一个类别视作正例,其余各类别视作负例,在概率预测矩阵中,按当前类别列下的概率值进行排序,随之调整划分阈值,可得到一条ROC曲线,如此,共可得到K条ROC曲线,对各曲线的坐标点求平均,即绘制出一条“宏计算(macro)”的ROC曲线,并可计算出它对应的AUC值。简单地说就是,概率预测矩阵的每一列值排序后都形成一条ROC曲线,再得到平均后的曲线。
方式二:
对于K(K>2)类分类问题,首先将样本标签one-hot化,得到一个K列的0和1标识的矩阵,将概率预测矩阵和该矩阵分别按行或按列将各自拼接为一维向量,按此概率向量进行排序,随之调整划分阈值,可得到一条“微计算(micro)”的ROC曲线,并可计算出它对应的AUC值。简单地说就是,概率预测矩阵的所有值排序后直接地形成一条ROC曲线。
4.Kappa系数
Kappa的计算与二分类是一致的,假设样本类别数为K,总样本数为N,实际为i预测为j的样本数为
a
i
j
a_{ij}
aij,有
k
a
p
p
a
=
p
o
−
p
e
1
−
p
e
kappa=\frac{p_o-p_e}{1-p_e}
kappa=1−pepo−pe
其中,
p
o
=
1
N
∑
i
=
1
K
a
i
i
p_o=\frac{1}{N}\sum_{i=1}^{K}{a_{ii}}
po=N1∑i=1Kaii是整体准确率,
p
e
=
1
N
2
∑
i
=
1
K
a
i
+
⋅
a
+
i
p_e=\frac{1}{N^2}\sum_{i=1}^{K}{a_{i+}\cdot a_{+i}}
pe=N21∑i=1Kai+⋅a+i 表偶然一致性。
例如,在某分类问题中,样本真实类别和预测类别分别是
y_true=[‘A’,‘A’,‘C’,‘B’,‘B’,‘A’,‘C’,‘B’];
y_pred=[‘C’,‘A’,‘C’,‘B’,‘B’,‘A’,‘B’,‘B’],
其对应的混淆矩阵为
则 p o = 2 + 3 + 1 8 = 0.75 p_o=\frac{2+3+1}{8}=0.75 po=82+3+1=0.75, p e = 3 × 2 + 3 × 4 + 2 × 2 8 × 8 = 0.3438 p_e=\frac{3\times 2+3\times 4+2\times 2}{8\times 8}=0.3438 pe=8×83×2+3×4+2×2=0.3438
所以 k a p p a = 0.75 − 0.3438 1 − 0.3438 = 0.619 kappa=\frac{0.75-0.3438}{1-0.3438}=0.619 kappa=1−0.34380.75−0.3438=0.619
四、Python实现
二分类情形
import numpy as np
from sklearn.tree import DecisionTreeClassifier
from sklearn import metrics
import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
from sklearn.metrics import cohen_kappa_score
## 1.定义数据集
# 训练数据
train_x = [
[4.8,3,1.4,0.3],
[5.1,3.8,1.6,0.2],
[4.6,3.2,1.4,0.2],
[5.3,3.7,1.5,0.2],
[5,3.3,1.4,0.2],
[7,3.2,4.7,1.4],
[6.4,3.2,4.5,1.5],
[6.9,3.1,4.9,1.5],
[5.5,2.3,4,1.3],
[6.5,2.8,4.6,1.5]
]
# 训练数据标签
train_y = [
'A',
'A',
'A',
'A',
'A',
'B',
'B',
'B',
'B',
'B'
]
# 测试数据
test_x = [
[3.1,3.5,1.4,0.2],
[4.9,3,1.4,0.2],
[5.1,2.5,3,1.1],
[6.2,3.6,3.4,1.3]
]
# 测试数据标签
test_y = [
'A',
'A',
'B',
'B'
]
train_x = np.array(train_x)
train_y = np.array(train_y)
test_x = np.array(test_x)
test_y = np.array(test_y)
## 2.训练分类器
clf_dt = DecisionTreeClassifier(max_depth=None, min_samples_split=2) # 定义决策树学习器
rclf_dt = clf_dt.fit(train_x,train_y) # 训练
## 3.数据计算
pre_y = rclf_dt.predict(test_x)
pre_y_proba = rclf_dt.predict_proba(test_x)[:,0] # 预测为类'A'的概率
## 4.性能评价
# (1) 准确率
accuracy = metrics.accuracy_score(test_y,pre_y)
# (2) 精确度(查准率)
precision = metrics.precision_score(test_y,pre_y,pos_label='A') # pos_label指定正例类
# (3) 召回率(查全率)
recall = metrics.recall_score(test_y,pre_y,pos_label='A')
# (4) F1度量
F1 = metrics.f1_score(test_y,pre_y,pos_label='A')
# (5) ROC-AUC值
fpr,tpr,threshold = metrics.roc_curve(test_y,pre_y_proba,pos_label='A')
AUC = metrics.auc(fpr,tpr)
# (6) Kappa系数
kappa = cohen_kappa_score(test_y, pre_y)
## 5.结果输出
print('\naccuracy:', accuracy)
print('\nprecision:', precision)
print('recall:', recall)
print('F1:', F1)
print('\nAUC:', AUC)
print('\nkappa:', kappa)
# ROC曲线绘图
plt.figure()
plt.plot(fpr, tpr, label='ROC curve (area = %0.2f)' % AUC)
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('Binary Classify ROC')
plt.legend(loc="lower right")
plt.show()
多分类情形
import numpy as np
from sklearn.tree import DecisionTreeClassifier
from sklearn import metrics
from sklearn.preprocessing import OneHotEncoder
from sklearn.metrics import cohen_kappa_score
## 1.定义数据集
# 训练数据
train_x = [
[4.8,3,1.4,0.3],
[5.1,3.8,1.6,0.2],
[4.6,3.2,1.4,0.2],
[5.3,3.7,1.5,0.2],
[5,3.3,1.4,0.2],
[7,3.2,4.7,1.4],
[6.4,3.2,4.5,1.5],
[6.9,3.1,4.9,1.5],
[5.5,2.3,4,1.3],
[6.5,2.8,4.6,1.5],
[6.3,3.3,6,2.5],
[5.8,2.7,5.1,1.9],
[7.1,3,5.9,2.1],
[6.3,2.9,5.6,1.8],
[6.5,3,5.8,2.2]
]
# 训练数据标签
train_y = [
'A',
'A',
'A',
'A',
'A',
'B',
'B',
'B',
'B',
'B',
'C',
'C',
'C',
'C',
'C'
]
# 测试数据
test_x = [
[5.1,3.5,1.4,0.2],
[4.9,3,1.4,0.2],
[5.1,2.5,3,1.1],
[5.7,2.8,4.1,1.3],
[6.2,3.4,5.4,2.3],
[5.9,3,5.1,1.8]
]
# 测试数据标签
test_y = [
'A',
'A',
'C',
'B',
'C',
'C'
]
train_x = np.array(train_x)
train_y = np.array(train_y)
test_x = np.array(test_x)
test_y = np.array(test_y)
## 2.训练分类器
clf_dt = DecisionTreeClassifier(max_depth=None, min_samples_split=2) # 定义决策树学习器
rclf_dt = clf_dt.fit(train_x,train_y) # 训练
## 3.数据计算
pre_y = rclf_dt.predict(test_x)
pre_y_proba = rclf_dt.predict_proba(test_x)
## 4.性能评价
# (1) 准确率
accuracy = metrics.accuracy_score(test_y,pre_y)
# (2) P-R度量
precision_macro = metrics.precision_score(test_y,pre_y,average='macro') #宏精确度(查准率)
recall_macro = metrics.recall_score(test_y,pre_y,average='macro') #宏召回率(查全率)
F1_macro = metrics.f1_score(test_y,pre_y,average='macro') #宏F1度量
precision_micro = metrics.precision_score(test_y,pre_y,average='micro') #微精确度(查准率)
recall_micro = metrics.recall_score(test_y,pre_y,average='micro') #微召回率(查全率)
F1_micro = metrics.f1_score(test_y,pre_y,average='micro') #微F1度量
# (3)ROC-AUC值
encoder = OneHotEncoder(sparse_output=False) # 创建一个OneHotEncoder对象
onehot_test_y = encoder.fit_transform(test_y.reshape(-1, 1)) # 对整数数组进行one-hot编码
AUC_macro = metrics.roc_auc_score(onehot_test_y,pre_y_proba, average='macro', sample_weight=None)
AUC_micro = metrics.roc_auc_score(onehot_test_y,pre_y_proba, average='micro', sample_weight=None)
# (4) Kappa系数
kappa = cohen_kappa_score(test_y, pre_y)
## 5.结果输出
print('\naccuracy: ', accuracy)
print('\nprecision_macro: ',precision_macro)
print('recall_macro:',recall_macro)
print('F1_macro:',F1_macro)
print('\nprecision_micro:',precision_micro)
print('recall_micro:',recall_micro)
print('F1_micro:',F1_micro)
print('\nAUC_macro: ',AUC_macro)
print('AUC_micro:',AUC_micro)
print('\nKappa: ',kappa)
End.
参考:
周志华. 机器学习. 清华大学出版社.
全文pdf下载