R ROC 곡선 겹치기

1 개요

R ROC 곡선 겹치기
set.seed(42) # 랜덤값 고정

df = infert
df = df[,c("age","parity","induced","spontaneous","case")]
# 데이터 분할
library(caret, quietly=T)
idx = createDataPartition(df$case, list=F, p=0.8)
Train = df[ idx,]
Test  = df[-idx,]

library(neuralnet)
model1 = neuralnet(case ~ age + parity + induced + spontaneous, Train, hidden=2)
Test$pred1 = predict(model1, Test, type="class")

library(C50)
Train$case = factor(Train$case)
model2 = C5.0(case ~ age + parity + induced + spontaneous, Train)
Test$pred2 = predict(model2, Test, type="prob")[,2]

library(ROCR, warn.conflicts=F)
pred1 = ROCR::prediction(Test$pred1, Test$case)
pred2 = ROCR::prediction(Test$pred2, Test$case)
pref1 = performance(pred1, "tpr", "fpr")
pref2 = performance(pred2, "tpr", "fpr")
plot(pref1, col="red")
plot(pref2, col="blue", add=T)
2 같이 보기
