[통계학] ROC 곡선 그리기

0. 목표

 - ROC 곡선 그리기

1. 실습

 1) library 호출

import pandas as pd
import matplotlib.pyplot as plt

 2) 데이터 생성

index = [i for i in range(1, 21)]
label = ['p', 'p', 'n', 'p', 'p', 'p', 'n', 'n', 'p', 'n', 'p', 'n', 'p', 'n', 'n', 'n', 'p', 'n', 'p', 'n']
probability = [0.9, 0.8, 0.7, 0.6, 0.55, 0.54, 0.53, 0.52, 0.51, 0.505, 0.4, 0.39, 0.38, 0.37, 0.36, 0.35, 0.34, 0.33, 0.3, 0.1]

 3) 데이터 프레임화

data = pd.DataFrame({'index' : index, 'label' : label, 'probability' : probability})

 - 데이터 결과

 4) TPR, FPR 계산 모듈

# inp1 : data
def ROC(inp1):
    FPR = []
    TPR = []
    P = len(inp1[inp1['label'] == 'p'])
    N = len(inp1[inp1['label'] == 'n'])
    for i in inp1['probability']:
        tmp_p = data[data['probability'] >= i]
        TP = len(tmp_p[tmp_p['label'] == 'p'])
        tmp_TPR = TP/P
        tmp_n = data[data['probability'] >= i]
        FP = len(tmp_n[tmp_n['label'] == 'n'])
        tmp_FPR = FP/N
        TPR.append(tmp_TPR)
        FPR.append(tmp_FPR)
    return TPR, FPR

 5) TPR, FPR 계산

TPR, FPR = ROC(data)

 6) 그래프 제작

fig = plt.figure()
fig.set_size_inches(15, 15)
plt.plot(FPR, TPR)
plt.xlabel("FPR", fontsize = 24)
plt.ylabel("TPR", fontsize = 24)
for i in range(len(data['probability'])):
    plt.text(FPR[i], TPR[i], data['probability'][i], fontsize = 18)

  - 결과

2. 참고

 - 데이터 과학자와 데이터 엔지니어를 위한 인터뷰 문답집(Hulu 데이터 과학팀)

 - matplotlib.pyplot.text