Bezdepbonusu7Программирование: статьи и ответы
Меню

Вопрос

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

Автор вопроса: KermitОтветов: 1

Я использую пакет precrec для оценки нескольких моделей и построения графиков ROC и PR-ROC.

В конце я хочу провести сравнение моделей, но, похоже, я не могу построить обе модели на одном графике.

Вот моя попытка:

library(precrec)
library(caret)
library(e1071)


classifier = svm(formula = Class ~ .,
                 data = train_smote_maison,
                 type = 'C-classification',
                 kernel = "linear",
                 probability = TRUE,
                 cross = 3,
                 cost = 1)

test_svm_plot = df[train.test.split == 2,]
predictions_svm2 <- predict(classifier,newdata = test_svm_plot, probability=T)
svm2_predict_obj <- mmdata(as.numeric(predictions_svm2),test_svm_plot$Class)
svm2_perfromance <- evalmod(svm2_predict_obj)


classifier_logreg <- glm(data = train, family = "binomial",
                         formula = Class ~ .)

test_glm = test
test_glm_plot = df[train.test.split == 2,]
predictions_logreg <- predict(classifier_logreg,newdata = test_glm, type = "response")
logreg_predict_obj <- mmdata(predictions_logreg,test_glm$Class)
logreg_performance <- evalmod(mdat = logreg_predict_obj)


plot(svm2_perfromance, "ROC")
plot(logreg_performance, "ROC", add=TRUE, col='red')

Кто-нибудь знает, как убедиться, что я могу получить оба ROC на одном и том же участке?

Заранее спасибо.

rlogistic-regressionsvmroce1071

Ответы

Ответ 1 · рейтинг 1 · katkoler

Вы можете использовать "fortify", чтобы создать кадр данных ggplot для каждого объекта производительности, затем объединить (rbind) их и построить с помощью ggplot.

library(ggplot2)

svm2_df <- fortify(svm2_perfromance)
logreg_df <- fortify(logreg_performance)

svm2_df$classifier <- "svm2"
logreg_df$classifier <- "logreg"

performance_df <- rbind(svm2_df, logreg_df)

roc <- performance_df[performance_df$classifier == "ROC",]

ggplot(roc, aes(x=x, y=y, group = classifier)) + geom_line(aes(color = classifier))

Перевод вопроса и ответов сообщества Stack Overflow, лицензия CC BY-SA. Оригинал на Stack Overflow.

Похожие материалы