热线电话:13121318867

登录
2018-10-26 阅读量: 939
ROC曲线如何绘制

一条曲线的绘制必须是多个点的连线,那么我们就需要在坐标轴上描出多个点,以此来绘制ROC曲线,具体步骤如下:

(1)假设我们有一个100行的数据集,按照9:1划分为训练集和测试集;

(2)接下来我们建立二分类模型并预测结果,这时10行的测试集中每个观测都应该有一个对应的原始分类和预测概率;

(3)这时我们依次将10个预测概率作为阈值threshold,当准确度大于或等于这个值的时候,预测结果为P,否则为N,例如,预测概率值为0.6,那么测试集结果为:

由此绘制混淆矩阵:

计算TPR与FPR:

TPR = TP/(TP+FN) = 0.3/(0.3+0.3) = 0.5

FPR = FP/(FP+TN) = 0.3/(0.1+0.3) = 0.75

据此,我们每更换一次阈值threshold,就能计算出一组TPR与FPR,那么这个数据集就应该能计算出10组TPR与FPR(其余9组请同学自行推算,不再写出具体计算过程)

(4)由此,我们以FPR为横坐标,TPR为纵坐标,描出所有的坐标点并连线,最终汇出ROC曲线,这里给出ROC曲线样例如下:

0.0000
2
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子