利用python画出AUC曲线的实例


Posted in Python onFebruary 28, 2020

以load_breast_cancer数据集为例,模型细节不重要,重点是画AUC的代码。

直接上代码:

from sklearn.datasets import load_breast_cancer
from sklearn import metrics
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import train_test_split
import pylab as plt
import warnings;warnings.filterwarnings('ignore')
dataset = load_breast_cancer()
data = dataset.data
target = dataset.target
X_train,X_test,y_train,y_test = train_test_split(data,target,test_size=0.2)
rf = RandomForestClassifier(n_estimators=5)
rf.fit(X_train,y_train)
pred = rf.predict_proba(X_test)[:,1]
#############画图部分
fpr, tpr, threshold = metrics.roc_curve(y_test, pred)
roc_auc = metrics.auc(fpr, tpr)
plt.figure(figsize=(6,6))
plt.title('Validation ROC')
plt.plot(fpr, tpr, 'b', label = 'Val AUC = %0.3f' % roc_auc)
plt.legend(loc = 'lower right')
plt.plot([0, 1], [0, 1],'r--')
plt.xlim([0, 1])
plt.ylim([0, 1])
plt.ylabel('True Positive Rate')
plt.xlabel('False Positive Rate')
plt.show()

利用python画出AUC曲线的实例

补充拓展:Python机器学习中的roc_auc曲线绘制

废话不多说,直接上代码

from sklearn.metrics import roc_curve,auc

from sklearn.ensemble import RandomForestClassifier

import matplotlib.pyplot as plt

from sklearn.model_selection import train_test_split

x_train,y_train,x_test,y_test=train_test_split(x,y,test_size=0.2)

rf=RandomForestClassifier()

rf.fit(x_train,y_train)

rf.score(x_train,y_train)

print('trainscore:'+str(rfbest.score(x_train,y_train)))
print('testscore:'+str(rfbest.score(x_test,y_test)))

y_score=rfbest.fit(x_train,y_train).predict_proba(x_test) #descision_function()不可用

print(type(y_score))

fpr,tpr,threshold=roc_curve(y_test,y_score[:, 1])
roc_auc=auc(fpr,tpr)
plt.figure(figsize=(10,10))
plt.plot(fpr, tpr, color='darkorange',
lw=2, label='ROC curve (area = %0.2f)' % roc_auc) ###假正率为横坐标,真正率为纵坐标做曲线
plt.plot([0, 1], [0, 1], color='navy', lw=2, linestyle='--')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('Receiver operating characteristic example')
plt.legend(loc="lower right")
plt.show()

以上这篇利用python画出AUC曲线的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python2.5/2.6实用教程 入门基础篇
Nov 29 Python
Python多线程编程(六):可重入锁RLock
Apr 05 Python
python 循环遍历字典元素的简单方法
Sep 11 Python
Python及PyCharm下载与安装教程
Nov 18 Python
用Python将一个列表分割成小列表的实例讲解
Jul 02 Python
python绘制简单彩虹图
Nov 19 Python
如何使用python操作vmware
Jul 27 Python
Python协程 yield与协程greenlet简单用法示例
Nov 22 Python
Python在终端通过pip安装好包以后在Pycharm中依然无法使用的问题(三种解决方案)
Mar 10 Python
利用python3筛选excel中特定的行(行值满足某个条件/行值属于某个集合)
Sep 04 Python
python 生成正态分布数据,并绘图和解析
Dec 21 Python
Python的flask接收前台的ajax的post数据和get数据的方法
Apr 12 Python
Python编程快速上手——选择性拷贝操作案例分析
Feb 28 #Python
AUC计算方法与Python实现代码
Feb 28 #Python
Python编程快速上手——Excel表格创建乘法表案例分析
Feb 28 #Python
Python计算IV值的示例讲解
Feb 28 #Python
Python编程快速上手——PDF文件操作案例分析
Feb 28 #Python
Python自动采集微信联系人的实现示例
Feb 28 #Python
python代码实现TSNE降维数据可视化教程
Feb 28 #Python
You might like
探讨:web上存漏洞及原理分析、防范方法
2013/06/29 PHP
国外十大最流行的PHP框架排名
2013/07/04 PHP
ThinkPHP之getField详解
2014/06/20 PHP
php实现xml转换数组的方法示例
2017/02/03 PHP
PHP命名空间定义与用法实例分析
2019/08/14 PHP
javascript移动设备Web开发中对touch事件的封装实例
2014/06/05 Javascript
js点击列表文字对应该行显示背景颜色的实现代码
2015/08/05 Javascript
javascript的replace方法结合正则使用实例总结
2016/06/16 Javascript
JavaScript九九乘法口诀表的简单实现
2016/10/04 Javascript
loading动画特效小结
2017/01/22 Javascript
浅谈Vue父子组件和非父子组件传值问题
2017/08/22 Javascript
微信小程序列表渲染功能之列表下拉刷新及上拉加载的实现方法分析
2017/11/27 Javascript
JS实现按比例缩小图片宽高
2020/08/24 Javascript
Python中lambda的用法及其与def的区别解析
2014/07/28 Python
python内存管理分析
2015/04/08 Python
web.py 十分钟创建简易博客实现代码
2016/04/22 Python
Python序列化基础知识(json/pickle)
2017/10/19 Python
Python AES加密实例解析
2018/01/18 Python
django 修改server端口号的方法
2018/05/14 Python
centos 安装Python3 及对应的pip教程详解
2019/06/28 Python
opencv python Canny边缘提取实现过程解析
2020/02/03 Python
python实现发送QQ邮件(可加附件)
2020/12/23 Python
CSS3新属性transition-property transform box-shadow实例学习
2013/06/06 HTML / CSS
澳大利亚时尚前卫设计师珠宝在线:Amber Sceats
2017/10/04 全球购物
Java程序员面试90题
2013/10/19 面试题
高中毕业自我鉴定
2013/12/13 职场文书
质检部部长职责
2013/12/16 职场文书
道德模范先进事迹
2014/02/14 职场文书
工程采购员岗位职责
2014/03/09 职场文书
爱之链教学反思
2014/04/30 职场文书
人事主管岗位职责说明书
2014/07/30 职场文书
给校长的一封检讨书
2014/09/20 职场文书
安全生产工作汇报
2014/10/28 职场文书
处罚决定书范文
2015/06/24 职场文书
2015年小学教师培训工作总结
2015/07/21 职场文书
如果用一句诗总结你的上半年,你会用哪句呢?
2019/07/16 职场文书