Кривая ROC с использованием вероятности прогнозируемого класса

Мне нужно нарисовать кривую ROC, используя предсказанные вероятности для задачи двух классов. Необходимо использовать различные отсечки вероятностей для построения ROC-кривой.

Я предсказываю вероятности класса, используя случайный лес

mydata<-read.table(file="out-all-gm-pr-hpcuts-wor-noAl.tr", header=TRUE, sep ="")
mydata$class <- as.factor(mydata$class)
mydata.rf<-randomForest(class ~ ., data=mydata,  importance = TRUE, mtry = 3, ntree = 100, proximity = TRUE )

Прогноз на тестовых данных с использованием вышеуказанного леса

mytestdata<-read.table(file="gmsim-craboff.tes",header=TRUE)
testpred<-predict(mydata.rf,mytestdata,type='prob')

Теперь у меня есть файл данных с истинной меткой класса и прогнозируемыми вероятностями для тестовых данных. Мне нужно сгенерировать кривую ROC, используя разные значения отсечки (скажем, 0.1, 0.3, 0.5, 0.7, 0.9) в вероятностях. Как это сделать?


r roc
person mradul    schedule 11.02.2013    source источник


Ответы (1)


Я бы сделал следующее:

library(pROC)
roc(mytestdata$class, testpred, plot = TRUE)
person Calimo    schedule 07.01.2014