快捷導(dǎo)航

python如何將多個模型的ROC曲線繪制在一張圖(含圖例)

更新時間：2022年02月24日 10:13:05 作者：柳奈奈

這篇文章主要給大家介紹了關(guān)于python如何將多個模型的ROC曲線繪制在一張圖的相關(guān)資料,文中通過實例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

多條ROC曲線繪制函數(shù)

 def multi_models_roc(names, sampling_methods, colors, X_test, y_test, save=True, dpin=100):
        """
        將多個機器模型的roc圖輸出到一張圖上
        
        Args:
            names: list, 多個模型的名稱
            sampling_methods: list, 多個模型的實例化對象
            save: 選擇是否將結(jié)果保存（默認(rèn)為png格式）
            
        Returns:
            返回圖片對象plt
        """
        plt.figure(figsize=(20, 20), dpi=dpin)

        for (name, method, colorname) in zip(names, sampling_methods, colors):
            
            method.fit(X_train, y_train)
            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)

        if save:
            plt.savefig('multi_models_roc.png')
            
        return plt

繪制效果

調(diào)用格式與方法

調(diào)用方法時，需要把模型本身（如clf_xx）、模型名字(如GBDT)和對應(yīng)顏色（如crimson）按照順序、以列表形式傳入函數(shù)作為參數(shù)。

names = ['Logistic Regression',
         'Random Forest',
         'XGBoost',
         'AdaBoost',
         'GBDT',
         'LGBM']

sampling_methods = [clf_lr,
                    clf_rf,
                    clf_xgb,
                    clf_adb,
                    clf_gbdt,
                    clf_lgbm
                   ]

colors = ['crimson',
          'orange',
          'gold',
          'mediumseagreen',
          'steelblue', 
          'mediumpurple'  
         ]

#ROC curves
train_roc_graph = multi_models_roc(names, sampling_methods, colors, X_train, y_train, save = True)
train_roc_graph.savefig('ROC_Train_all.png')

詳細(xì)解釋和說明

1.關(guān)鍵函數(shù)

（1）plt.figure(figsize=(20, 20), dpi=dpin)

在for循環(huán)外繪制圖片的大體框架。figsize控制圖片大小，dpin控制圖片的信息量（其實可以理解為清晰度？documentation的說明是The resolution of the figure in dots-per-inch）

（2）zip()

函數(shù)用于將可迭代的對象作為參數(shù)，將對象中對應(yīng)的元素打包成一個個元組，然后返回由這些元組組成的列表。

（3）roc_curve()

fpr, tpr, thresholds = roc_curve(y_test, y_test_predprob, pos_label=1)

該函數(shù)的傳入?yún)?shù)為目標(biāo)特征的真實值y_test和模型的預(yù)測值y_test_predprob。需要為pos_label賦值，指明正樣本的值。

該函數(shù)的返回值 fpr、tpr和thresholds 均為ndarray，為對應(yīng)每一個不同的閾值下計算出的不同的真陽性率和假陽性率。這些值，就對應(yīng)著ROC圖中的各個點。

（4）auc()

 plt.plot(fpr, tpr, lw=5, label='{} (AUC={:.3f})'.format(name, auc(fpr, tpr)),color = colorname)

函數(shù)auc()，傳入?yún)?shù)為fpr和tpr，返回結(jié)果為模型auc值，即曲線下面積值。

以上代碼在使用fpr和tpr繪制ROC曲線的同時，也確定了標(biāo)簽（圖例）的內(nèi)容和格式。

2. 參數(shù)解釋

（1）sampling_methods

是包含多個模型名字的list。所有模型不需要fit過再傳入函數(shù)，只需要定義好即可。

clf = RandomForestClassifier(n_estimators = 100, max_depth=3, min_samples_split=0.2, random_state=0)

（2）X_test, y_test

X_test 和 y_test 兩個參數(shù)用于傳入函數(shù)后計算各個模型的預(yù)測值。

y_test_predprob = method.predict_proba(X_test)[:,1]
fpr, tpr, thresholds = roc_curve(y_test, y_test_predprob, pos_label=1)

如果需要繪制的是訓(xùn)練集的ROC曲線，則可以在對應(yīng)參數(shù)位置分別傳入X_trian和y_train即可。

（3）names 和 colors

這兩個參數(shù)均為字符串列表形式。注意，這兩個列表的值要和模型參數(shù)中的模型順序一一對應(yīng)。

如有需要繪制更多的模型，只需要對應(yīng)增加列表中的值即可。

需要注意的小小坑

1.同一張圖片的同一種方法只能調(diào)用一次?。?！

plt.legend(loc='lower right')
plt.legend(fontsize=10)

如果像上圖中的我一樣，把同一張圖片plt的方法legend()調(diào)用兩次，那么下一個的方法中的參數(shù)就會將上一個的參數(shù)覆蓋！這種情況下，我就發(fā)現(xiàn)第一個方法賦值的location完全不起作用……

這個時候就需要將這個函數(shù)整合如下圖～（其實本來就是應(yīng)該這么寫的，我也不知道為啥我腦子一抽寫了兩個，可能是ggplot給我的美好印象揮之不去吧）

plt.legend(loc='lower right',fontsize=10)

補充

根據(jù)小伙伴的評論提問，在這里進行一下解釋說明：

1.這個函數(shù)是適用于所有數(shù)據(jù)集的，只需要導(dǎo)入數(shù)據(jù)集后進行訓(xùn)練集和測試集的劃分即可。（我在“調(diào)用格式與方法”部分調(diào)用函數(shù)使用的是X_train 和y_train，繪制出的則是不同模型在訓(xùn)練集表現(xiàn)的ROC曲線）

劃分訓(xùn)練集和測試集的代碼如下（以使用8：2劃分訓(xùn)練集測試集為例）

# 8：2劃分訓(xùn)練集測試集
X, y = df.drop(target,axis=1), df[target]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.8, random_state=0)

df：導(dǎo)入數(shù)據(jù)集

target：目標(biāo)特征（y）

train_size：訓(xùn)練集占比80%

random_state: 隨機數(shù)種子，不同隨機數(shù)種子劃分的訓(xùn)練集和測試集會有不同。