Programing/R- programming

R에서 ROC curve 그리기

sosal 2018. 3. 19. 01:09
반응형

/*

 http://sosal.kr/
 * made by so_Sal
 */


R 프로그래밍에서 Logistic regression, SVM, 혹은 딥러닝 결과 등을 정리할 때

ROC curve와 AUC를 계산하기 위해서 매번  pROC, ROCR 등의 라이브러리로

허접한 기본 figure들을 만들었었다.



항상 찝찝한 마음으로 기본적인 figure를 그리다가..

오늘 찾아보니 Epi라는 패키지를 쓰면 매우 쉽게 좋은 ROC curve 그래프를 그릴 수 있다는 것을 발견했다.



install.packages("Epi")

간단하게 설치해주고..


library(Epi)
ROC(predicted, label) # also try adding plot="sp"

코드 한줄이면 그래프가 그려진다.




ex)


library(Epi)


predicted <- c( rnorm(500, 0.35, 0.1), rnorm(500, 0.65, 0.11) )

label <- c( rep(0, 500), rep(1, 500))
ROC(predicted, label)