Формат прогноза для кривых ROCR в R

Question

Я впервые приближаюсь к ROCR-кривым в R и использую

#ROCR Curve
library(ROCR)

predict_rf <- predict(Model_RF, type = 'response')

ROCRpred_rf <- prediction(predict_rf, test.5$si2014)
ROCRperf_rf <- performance(ROCRpred_rf, 'tpr','fpr')
plot(ROCRperf_rf, colorize = TRUE, text.adj = c(train-0.2,1.7))

Я получил

> #ROCR Curve
> library(ROCR)
> 
> predict_rf <- predict(Model_RF, type = 'response')
> 
> ROCRpred_rf <- prediction(predict_rf, test.5$si2014)
Error in prediction(predict_rf, test.5$si2014) : 
  Format of predictions is invalid.

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

> sapply(test.5, class)
   spesa_tot     n_visite      importo        sesso eta_abbonati 
   "numeric"    "numeric"    "numeric"     "factor"    "numeric" 
      si2014     mesi_tot    Residenza      cluster 
   "numeric"    "integer"     "factor"     "factor" 

В чем проблема?С моделью SVM у меня вообще нет проблем ....

Answer 1

Я решил это.Это было легко.Просто сделайте

predict_rf <- predict(Model_RF, type = 'response')
predict_rf
ROCRpred_rf <- prediction(as.numeric(predict_rf), as.numeric(train.4$si2014))
ROCRperf_rf <- performance(ROCRpred_rf, 'tpr','fpr')
plot(ROCRperf_rf, colorize = TRUE, text.adj = c(train.4-0.2,1.7))

Я изменил

ROCRpred_rf <- прогноз (предикат, тест.5 $ si2014) </em>

с

ROCRpred_rf <- прогноз (как числовой (предикат), числовой (поезд.4 $ si2014)) </strong>