ROC 곡선과 AUC 출력

파이썬으로 ROC 곡선을 그려 보았습니다.



전제


  • Python
  • 로지스틱 회귀에서 예측값 출력됨
  • scikit-learn과 matplotlib 사용

  • 코드



    roc.py
    from sklearn import metrics
    import matplotlib.pyplot as plt
    import numpy as np
    
    # FPR, TPR(, しきい値) を算出
    fpr, tpr, thresholds = metrics.roc_curve(test_y, predict_y)
    
    # ついでにAUCも
    auc = metrics.auc(fpr, tpr)
    
    # ROC曲線をプロット
    plt.plot(fpr, tpr, label='ROC curve (area = %.2f)'%auc)
    plt.legend()
    plt.title('ROC curve')
    plt.xlabel('False Positive Rate')
    plt.ylabel('True Positive Rate')
    plt.grid(True)
    



    참고



    ROC 곡선과 AUC에 대해서는 이쪽을 참고로.

    【통계학】 ROC 곡선이란 무엇인가, 애니메이션으로 이해한다.
    【ROC 곡선과 AUC】기계 학습의 평가 지표에 대한 기초 강좌

    좋은 웹페이지 즐겨찾기