Как получить значения осей X (FPR) и Y (TPR) в кривой ROC? - PullRequest
Новая
       59

Как получить значения осей X (FPR) и Y (TPR) в кривой ROC?

3 голосов
/ 11 декабря 2019

Я пытаюсь создать кривую ROC, используя plotly.js со значениями, полученными в R.

В plotly, я должен заполнить значения (ось X, ось Y)чтобы построить график.

Но я не знаю, как получить значения при использовании функции логистической регрессии glm или multinom (в nnet пакетах). 

df <- data.frame(age = c(10, 20, 30, 40, 50),height = c(150, 161, 141, 155, 180), house = c("0", "0", "0", "1", "1"))
smp_size <- floor(nrow(df) * (0.6))
train_idx <- sample(seq_len(nrow(df)), size = smp_size)

train <- df[train_idx, ]
validation <- df[-train_idx, ]

validation_x <- validation[c("age", "height")]
validation_y <- validation[["house"]]

m <- glm(house ~ age + height, data = train, family = 'binomial')
y_pred <- predict(m, validation_x)

library(plotROC)
plotROC(validation_y, y_pred)

Я хочу знать, как получить значения оси X (FPR) и оси Y (TPR) на кривой ROC

спасибо.

...