Python中ROC曲线绘制
首先以支持向量机模型为例
先导入需要使用的包，我们将使用roc_curve这个函数绘制ROC曲线！
from sklearn.svm import SVC from sklearn.metrics import roc_curve from sklearn.datasets import make_blobs from sklearn. model_selection import train_test_split import matplotlib.pyplot as plt %matplotlib inline
然后使用下面make_blobs函数，生成一个二分类的数据不平衡数据集；
使用train_test_split函数划分训练集和测试集数据；
训练SVC模型。
X,y = make_blobs(n_samples=(4000,500), cluster_std=[7,2], random_state=0) X_train,X_test,y_train, y_test = train_test_split(X,y,random_state=0) clf = SVC(gamma=0.05).fit(X_train, y_train)
fpr,tpr, thresholds = roc_curve(y_test,clf.decision_function(X_test))
plt.plot(fpr,tpr,label='ROC')
plt.xlabel('FPR')
plt.ylabel('TPR')
从上面的代码可以看到，我们使用roc_curve函数生成三个变量，分别是fpr,tpr, thresholds，也就是假正例率（FPR）、真正例率（TPR）和阈值。
而其中的fpr,tpr正是我们绘制ROC曲线的横纵坐标，于是我们以变量fpr为横坐标，tpr为纵坐标，绘制相应的ROC图像如下：
值得注意的是上面的支持向量机模型使用的decision_function函数，是自己所特有的，而其他模型不能直接使用。
比如说我们想要使用其他模型（例如决策树模型）的结果绘制ROC,直接套用上面的代码，会报错，会显示没有这个函数。
以决策树模型为例，解决上述问题（适用于除向量机外的模型）
导入决策树模型包以及训练模型的代码省略了，只需要手动改一改就行了，我们直接看绘图的代码！
fpr,tpr, thresholds = roc_curve(y_test,clf.predict_proba(X_test)[:,1])
plt.plot(fpr,tpr,label='ROC')
plt.xlabel('FPR')
plt.ylabel('TPR')
可以看到我们直接把只适用于支持向量机模型的函数decision_function更改成predict_proba(X_test)[:,1]就行了，让我们看看结果：
可以看到哈，决策树模型在这个数据集上的泛化能力不如支持向量机哈！！！学废了吗。
我可以失败，但绝不会认输！
到此这篇关于Python中ROC曲线绘制的文章就介绍到这了,更多相关Python ROC曲线绘制内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了python类属性学习深入讲解，文中对于python的类属性的理解有正在学习python的同学可以一块学习下2021-03-03
利用pandas如何实现类似sql中的left join操作
这篇文章主要介绍了利用pandas如何实现类似sql中的left join操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-06-06
这篇文章主要介绍了Python爬虫实现验证码登录，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-05-05
今天小编就为大家分享一篇python写入并获取剪切板内容的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
Python输出列表(List)不带中括号和引号的问题及解决方法
这篇文章主要介绍了Python输出列表(List)不带中括号和引号的问题及解决方法，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2023-02-02
这篇文章主要介绍了Django的forms组件，本文通过实例代码介绍了Django的forms组件,需要的朋友可以参考下2018-11-11
在python新式类中，可以定义一个变量__slots__，它的作用是阻止在实例化类时为实例分配dict，下面这篇文章主要给大家介绍了关于Python中__slots__的相关资料，需要的朋友可以参考借鉴，下面来一起看看吧。2017-07-07
这篇文章主要为大家详细介绍了python网络爬虫学习笔记的第一篇，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-04-04
Python中使用haystack实现django全文检索搜索引擎功能
django是python语言的一个web框架，功能强大。配合一些插件可为web网站很方便地添加搜索功能。下面通过本文给大家分享Python中使用haystack实现django全文检索搜索引擎功能，感兴趣的朋友一起看看吧2017-08-08
今天小编就为大家分享一篇使用python的pyplot绘制函数实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
最新评论