首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >python对随机森林分类结果绘制roc曲线

python对随机森林分类结果绘制roc曲线

作者头像
全栈程序员站长
发布2022-08-29 13:58:04
发布2022-08-29 13:58:04
1.7K00
代码可运行
举报
运行总次数:0
代码可运行

大家好,又见面了,我是你们的朋友全栈君。

上图:

【注意!!!!! 以下代码用到的roc_curve函数(只能用于二分类),如果多分类会报错,不适用于多分类!!!!!】

附上代码:一个函数,传入三个参数

代码语言:javascript
代码运行次数:0
运行
复制
.....传入参数,训练模型,然后:
fit = model.fit(x_train, y_training)
# ROC
y_score = model.fit(x_train, y_training).predict_proba(x_test)  # 随机森林
fpr, tpr, thresholds = roc_curve(y_test, y_score[:, 1])
roc_auc = auc(fpr, tpr)
代码语言:javascript
代码运行次数:0
运行
复制
def drawRoc(roc_auc,fpr,tpr):
    plt.subplots(figsize=(7, 5.5))
    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('ROC Curve')
    plt.legend(loc="lower right")
    plt.show()
代码语言:javascript
代码运行次数:0
运行
复制
drawRoc(roc, fpr, tpr)
代码语言:javascript
代码运行次数:0
运行
复制
注:导入的包没有贴上,需要自己导入,复制代码即可运行

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/145587.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022年5月1,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档