Как построить кривую ROC для предсказания случайного леса классификации в R?

Это, наверное, простой вопрос, но я застрял на нем целую вечность. Я построил классификационную модель случайного леса, чтобы предсказать классификации моего тестового набора, и прикрепил результат к моему кадру данных следующим образом.

RandomTree <- randomForest(as.factor(OutcomeVariable) ~ ., data=TrainSet, ntree=30 , 
importance=TRUE, samplesize=c(5195,5195))
Prediction<-predict(RandomTree, TestSet, type="response")
TestSet$Prediction<-Prediction

Я хочу попытаться создать кривую ROC и рассчитать AUC на ее основе. Я понимаю, что прогнозы классификации необходимо преобразовать в вероятности, однако я не знаю, как это сделать. Причина этого в том, что я могу сравнить результаты моего случайного леса с логистической регрессией, которую я также смоделировал. Мне удалось получить кривую ROC из моей логистической регрессии.

Любая помощь приветствуется. Спасибо


person Fiona    schedule 10.08.2017    source источник
comment
Пакет caret может вам помочь. Он имеет функции, разработанные специально для этого (построение кривых ROC, вычисление AUC, сравнение моделей).   -  person Aurèle    schedule 10.08.2017
comment
Возможный дубликат кривой ROC для классификации из randomForest   -  person Calimo    schedule 10.08.2017