python画ROC曲线如何画的好看

import numpy as np
from sklearn.metrics import confusion_matrix, accuracy_score,f1_score,roc_auc_score,recall_score,precision_score
from sklearn import metrics
from matplotlib import pyplot


styles=['fivethirtyeight',
 'dark_background',
 'bmh',
 'classic',
 'seaborn-dark',
 'grayscale',
 'seaborn-deep',
 'seaborn-muted',
 'seaborn-colorblind',
 'seaborn-white',
 'seaborn-dark-palette',
 'ggplot',
 'tableau-colorblind10',
 '_classic_test',
 'seaborn-darkgrid',
 'seaborn-notebook',
 'Solarize_Light2',
 'seaborn-paper',
 'seaborn-whitegrid',
 'seaborn-pastel',
 'seaborn-talk',
 'seaborn-bright',
 'seaborn',
 'seaborn-ticks',
 'seaborn-poster',
 'fast']

def rocs_plot(y2, P2, setname=''):

    pyplot.figure(figsize=(5, 4), dpi=100)
    pyplot.style.use('seaborn-darkgrid')
    
    palette = pyplot.get_cmap('Set1')
    
        
    for i, p2 in enumerate(P2):
        fpr1, tpr1, thresholds1 = metrics.roc_curve(y2[i], p2, pos_label=1)
        roc_auc1 = metrics.auc(fpr1, tpr1)   
        pyplot.plot(fpr1, tpr1, lw=1, color=palette(i), linewidth=1.5, alpha=0.9, label=result_paths[i][:-10]+", AUC=%0.3f" % (roc_auc1))       
    pyplot.xlim([0.00, 1.0])
    pyplot.ylim([0.00, 1.0])
    pyplot.xlabel("1-Specificity",fontsize=12)
    pyplot.ylabel("Sensitivity",fontsize=12)
    pyplot.title("ROC",fontsize=12)
    pyplot.legend(loc="lower right")
    pyplot.savefig('ROC.png',dpi=300)

pyplot画图太丑,改个风格就好

pyplot.style.use('seaborn-darkgrid')

里面的风格可以自己改,'seaborn-darkgrid' 风格出来大概是这样的,比默认格式好看多了:

  • 0
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值