weixin_55054803 2022-04-30 04:21 采纳率: 75%
浏览 110
已结题

多模型roc曲线的绘制不出图

#码可以跑起来,但是不出图,劳烦各位看看哪里有问题
def multi_models_roc(names, sampling_methods, colors, X_test, y_test, save=True, dpin=100):

    plt.figure(figsize=(20, 20), dpi=dpin)

    for (name, method, colorname) in zip(names=['Logistic Regression',
                                                'Decision Tree',
                                                'SVM',
                                                'Random Forest',
                                                'XGBoost'
                                               ],
                                         sampling_methods =[clf_lr,
                                                            clf_tr,
                                                            clf_svm,
                                                            clf_forest,
                                                            clf_xgbc
                                                           ],
                                         colors=['crimson',
                                                 'orange',
                                                 'gold',
                                                 'mediumseagreen',
                                                 'steelblue'
                                                ]
                                         train_roc_graph = multi_models_roc(names, sampling_methods, colors, X_train, y_train, save = True)
                                         train_roc_graph.savefig('ROC_Train_all.png')
                                        ):
        y_test_preds = method.predict(X_test)
        y_test_predprob = method.predict_proba(X_test)[:,1]
        fpr, tpr, thresholds = roc_curve(y_test, y_test_predprob, pos_label=1)
        
        plt.plot(fpr, tpr, lw=5, label='{} (AUC={:.3f})'.format(name, auc(fpr, tpr)),color = colorname)
        plt.plot([0, 1], [0, 1], '--', lw=5, color = 'grey')
        plt.axis('square')
                 plt.xlim([0, 1])
        plt.ylim([0, 1])
        plt.xlabel('False Positive Rate',fontsize=20)
        plt.ylabel('True Positive Rate',fontsize=20)
        plt.title('ROC Curve',fontsize=25)
        plt.legend(loc='lower right',fontsize=20)
        plt.show()
        
    
    if save:
        plt.savefig('multi_models_roc.png')
        
    return plt

img

img

展开全部

  • 写回答

2条回答 默认 最新

  • 白驹_过隙 算法领域新星创作者 2022-04-30 04:41
    关注

    在代码第一行加上这个试试

    %matplotlib inline
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
    weixin_55054803 2022-04-30 05:04

    这个不行我试过了

    回复
    白驹_过隙 回复 weixin_55054803 2022-04-30 05:32

    其实你可以测试一个简单的画图看看是notebook 的原因还是代码的原因

    回复
    weixin_55054803 回复 白驹_过隙 2022-04-30 05:38

    代码的原因,我单个模型是能画出来的

    回复
    展开全部8条评论
查看更多回答(1条)
编辑
预览

报告相同问题?

问题事件

  • 系统已结题 5月7日
  • 已采纳回答 4月30日
  • 创建了问题 4月30日

悬赏问题

  • ¥15 为什么树莓派5b显示禁止连接
  • ¥20 流量太费!寻找便宜的app音视频SDK或平替方案。
  • ¥15 kubeasz部署遇到问题
  • ¥15 GUIDE to App Designer Migration Tool for MATLAB
  • ¥50 第三代非支配排序遗传算法(NSGA-Ⅲ)和多目标粒子群优化算法(MOPSO)的实现
  • ¥20 plant simulation与python com接口实时数据交互
  • ¥15 有关汽车的MC9S12XS128单片机实验
  • ¥15 求c语言动态链表相关课程有偿,或能将这块知识点讲明白
  • ¥15 FLKT界面刷新异常
  • ¥15 物体双站RCS和其组成阵列后的双站RCS关系验证
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部