如何在一张图中画不同模型的roc曲线（python）

xiaoshu_yilian

已于 2022-09-01 15:38:16 修改

阅读量823

点赞数

分类专栏： python 机器学习文章标签： python 机器学习人工智能

于 2022-07-05 11:43:38 首次发布

本文链接：https://blog.csdn.net/weixin_38003620/article/details/125616674

版权

import pandas as pd
import warnings
warnings.filterwarnings('ignore')

from   matplotlib import pyplot as plt
from sklearn.metrics import roc_auc_score, roc_curve,auc
from sklearn.ensemble import AdaBoostClassifier
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import BaggingClassifier
from sklearn.ensemble import ExtraTreesClassifier
from sklearn.model_selection import train_test_split as sp
from xgboost import XGBClassifier
from lightgbm import  LGBMClassifier
from sklearn import datasets
from sklearn.model_selection import StratifiedKFold
from sklearn.linear_model import LogisticRegression as LR

def multi_models_roc(names, sampling_methods, colors, train, test, save=True, dpin=100):
    plt.figure(figsize=(20, 20)