Python脚本实现ROC曲线图

最新推荐文章于 2025-04-18 15:28:33 发布

shandong_chu

最新推荐文章于 2025-04-18 15:28:33 发布

阅读量2.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/ShanDong_Chu/article/details/91043807

本文通过一个实例展示了如何利用Python脚本创建ROC曲线图。首先介绍了需要下载matplotlib和numpy库，然后提供了一个数据文件a.txt，接着详细解释了脚本a.py的内容，包括读取数据、绘制曲线、设置坐标轴标签和图表标题，并最终运行脚本得到ROC曲线图。通过分析ROC曲线的位置，可以判断算法的优劣。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

通过实例来讲解Python脚本来实现构建ROC曲线图。

步骤：

1、下载模块： matplotlib、numpy

2、准备数据：

例如：a.txt，内容如下

0.87,0.000001
0.87,0.000003
0.92,0.000006
0.93,0.000011
0.94,0.000017
0.94,0.000028
0.94,0.000044

3、编写脚本：a.py

#coding:utf-8
import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib.font_manager import FontProperties
import matplotlib as mpl
#指定默认字体
mpl.rcParams['font.sans-serif'] = ['SimHei']
mpl.rcParams['font.family']='sans-serif'
#解决负号'-'显示为方块的问题
mpl.rcParams['axes.unicode_minus'] = False

a_y, a_x = np.loadtxt('a.txt', delimiter=',', unpack=True)
plt.plot(a_x,a_y,label='a',color='green')
plt.plot(a_x,a_y,color='green')

plt.xlabel('反向概率(%)')
plt.ylabel('正向概率(%)')

plt.title(u'算法ROC对比')
plt.legend()
plt.show()

4、运行脚本

打开命令行窗口，执行：python a.py