将排名作为阈值做ROC曲线

金刚和娃儿

于 2021-10-18 19:36:15 发布

阅读量227

点赞数 1

分类专栏： ROC 文章标签： python

本文链接：https://blog.youkuaiyun.com/a_Mr_Da/article/details/120832912

版权

排名作为阈值绘制ROC曲线

有时我们的结果需要按照最终标签排名作出ROC曲线,如下代码那么我们只需要预测得分和真实标签值。

a.将真实标签按照得分排序
b.排序后的标签按照排名作为阈值,依次从1到最后,每个阈值得到一次FPR和TPR.

大家可以比较一下我写的对不对,如有问题欢迎指出. 代码片.

import matplotlib.pyplot as plt


class Rankauc:
    def __init__(self, score_list, label_list):
        self.score_list = score_list
        self.label_list = label_list
        self.len = len(label_list)
        self.F_num = label_list.count(0)
        self.T_num = label_list.count(1)

    def binding_sort