【发布时间】:2020-10-31 23:47:10
【问题描述】:
对于我知道所需输出的较小数据集,我也可以这样做
例如: 我有数据集:
y_true=[1,1,0,1]
y_pred=[1,0,0,0]
我可以编写函数来查找所有真假阳性和阴性:
def measures(y_pred, y_true):
TP = 0
FP = 0
TN = 0
FN = 0
for i in range(len(y_pred)):
if y_true[i]==y_pred[i]==1:
TP += 1
if y_pred[i]==1 and y_true[i]!=y_pred[i]:
FP += 1
if y_true[i]==y_pred[i]==0:
TN += 1
if y_pred[i]==0 and y_true[i]!=y_pred[i]:
FN += 1
return(TP, FP, TN, FN)
但是,如果我打算为多个标签找到相同的东西怎么办,像这样:
y_true=[1,2,0,1,3]
y_pred=[1,0,3,2,1]
所以我想找到所有的 TP_total、TN_total、FP_total 和 FN_total 考虑每个标签'1','2','0','3'
我该怎么做
【问题讨论】:
-
查看混淆矩阵,列代表实际类别,行代表预测类别
标签: python machine-learning scikit-learn classification