2018-10-26
閱讀量:
971
ROC曲線如何繪制
一條曲線的繪制必須是多個點(diǎn)的連線,那么我們就需要在坐標(biāo)軸上描出多個點(diǎn),以此來繪制ROC曲線,具體步驟如下:
(1)假設(shè)我們有一個100行的數(shù)據(jù)集,按照9:1劃分為訓(xùn)練集和測試集;
(2)接下來我們建立二分類模型并預(yù)測結(jié)果,這時10行的測試集中每個觀測都應(yīng)該有一個對應(yīng)的原始分類和預(yù)測概率;

(3)這時我們依次將10個預(yù)測概率作為閾值threshold,當(dāng)準(zhǔn)確度大于或等于這個值的時候,預(yù)測結(jié)果為P,否則為N,例如,預(yù)測概率值為0.6,那么測試集結(jié)果為:

由此繪制混淆矩陣:

計(jì)算TPR與FPR:
TPR = TP/(TP+FN) = 0.3/(0.3+0.3) = 0.5
FPR = FP/(FP+TN) = 0.3/(0.1+0.3) = 0.75
據(jù)此,我們每更換一次閾值threshold,就能計(jì)算出一組TPR與FPR,那么這個數(shù)據(jù)集就應(yīng)該能計(jì)算出10組TPR與FPR(其余9組請同學(xué)自行推算,不再寫出具體計(jì)算過程)
(4)由此,我們以FPR為橫坐標(biāo),TPR為縱坐標(biāo),描出所有的坐標(biāo)點(diǎn)并連線,最終匯出ROC曲線,這里給出ROC曲線樣例如下:







評論(0)


暫無數(shù)據(jù)
推薦帖子
0條評論
0條評論
0條評論