Кривая RO C с использованием PRROC - PullRequest
0 голосов
/ 10 апреля 2020

У меня есть двоичные данные покрытия (0/1), полученные из классификационной модели случайного леса. I sh для расчета кривой RO C с использованием наблюдаемых и прогнозируемых значений покрытия. Я не мог использовать пакет ROCR в моем Rstudio (R версия 3.4.4), поэтому я пробую PRRO C. Я получаю AU C 0,5, что не имеет смысла. Пожалуйста, подскажите, где у меня ошибка.

Я использовал следующий код:

library(PRROC)

rf_pred <- predict(RFmodel, train$loss, type="prob")
fg_rf <- rf_pred[train$loss==1]
fg_rf
bg_rf <- rf_pred[train1$loss==0]
bg_rf
roc <- roc.curve(scores.class0 =fg_rf, weights.class0=bg_rf, curve=TRUE)
plot(roc)
...