1. 학습
    2. /
  2. 강의
    3. /
  3. scikit-learn으로 배우는 Supervised Learning
Connected

연습 문제

ROC 곡선

이제 당뇨 여부를 예측하는 로지스틱 회귀 모델을 만들었으니, 결정 임계값이 바뀔 때 진양성률과 위양성률이 어떻게 달라지는지 시각화하기 위해 ROC 곡선을 그려 보세요.

테스트 레이블 y_test와 테스트 특성이 양성 클래스일 예측 확률 y_pred_probs가 미리 로드되어 있으며, matplotlib.pyplot은 plt로 임포트되어 있어요.

ROC 곡선을 만든 뒤, 결과를 해석해 보겠습니다.

지침 1/2

undefined XP
    1
    2
  • roc_curve를 임포트하세요.
  • y_test와 y_pred_probs를 사용해 ROC 곡선 값을 계산하고, 결과를 fpr, tpr, thresholds로 언패킹하세요.
  • 위양성률 대비 진양성률을 플로팅하세요.