将排名作为阈值做ROC曲线
排名作为阈值绘制ROC曲线
有时我们的结果需要按照最终标签排名作出ROC曲线,如下代码那么我们只需要预测得分和真实标签值。
a.将真实标签按照得分排序
b.排序后的标签按照排名作为阈值,依次从1到最后,每个阈值得到一次FPR和TPR.
大家可以比较一下我写的对不对,如有问题欢迎指出. 代码片.
import matplotlib.pyplot as plt
class Rankauc:
def __init__(self, score_list, label_list):
sel
原创
2021-10-18 19:36:15 ·
161 阅读 ·
0 评论