1. 학습
    2. /
  2. 강의
    3. /
  3. scikit-learn으로 배우는 지도 학습
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로 언패킹하세요.
  • 위양성률에 대한 진양성률을 그래프로 나타내세요.