多分类多标签的评价指标以及阈值调优

多分类多标签的评价指标以及阈值调优,第1张

多分类/多标签的评价指标以及阈值调优 多标签/多分类 准确率

sklearn.metrics.precision_score(y_true, y_pred, labels=None, pos_label=1, average=’binary’, sample_weight=None)
average参数定义了该指标的计算方法,二分类时average参数默认是binary,多分类时,可选参数有
micro: 整体准确率
macro: 平均的准确率,不考虑各种标签的权重
weighted: 带权重的准确率,考虑各种标签的权重
我理解weighted这种方式更加合理一些
https://zhuanlan.zhihu.com/p/59862986

召回率

计算召回率由相应的函数
sklearn.metrics.recall_score
https://blog.csdn.net/pearl8899/article/details/109874210

阈值调优

针对每一种标签单独选择阈值,使得P+R最大
1)使用sklearn.metrics.precision_recall_curve得到PR曲线
2)利用argmax函数求最优解
下面代码的目标为max(precision+recall)
*3)如果有一些限制条件,例如P>0.9,可以在得到PR曲线之后删掉一部分后再使用argmax函数
(不知道能否使用scipy,linego等专业工具解决该问题,或许没有必要)
(如果数据过多可能需要使用spark进行加速)

import numpy as np
from numpy.core.fromnumeric import argmax
from sklearn.metrics import precision_recall_curve
y_true = np.array([0, 0, 1, 1,0,0,1])
y_scores = np.array([0.3, 0.2, 0.65, 0.7, 0.4, 0.15, 0.8])
precision, recall, thresholds = precision_recall_curve(y_true, y_scores)

print(precision)
print(recall)
print(thresholds)
target = precision + recall
index = argmax(target)
print(thresholds[index])

sklearn.metrics.precision_recall_curve的源码解析如下
https://zhuanlan.zhihu.com/p/404798546

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/4667431.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-06
下一篇 2022-11-06

发表评论

登录后才能评论

评论列表(0条)

保存