2013-02-11 238 views
0

我需要使用两类问题的预测概率绘制ROC曲线。需要使用不同的截断概率来生成ROC曲线。使用预测类别的概率的ROC曲线

使用随机森林使用以上森林

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

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) 

试验数据预测我现在有一个数据文件,现在真正的类标签,我预测类的概率和预测的测试数据概率。我需要使用不同的临界值(如0.1,0.3,0.5,0.7,0.9)以概率生成ROC曲线。如何去做呢?

回答

0

我会做到以下几点:

library(pROC) 
roc(mytestdata$class, testpred, plot = TRUE)