¿Las probabilidades predichas de modelos glm son probabilidades de 0 o 1?
Mi variable de respuesta, status, tiene dos valores, 1 para vivo y 0 para muerto.
He construido un modelo como este modelo <- glm(status ~., train_data, family = ‘binomial’). Utilizo predict(model, test_data, type = ‘response’), lo que me da un vector de probabilidades predichas, como este:
0.02 0.04 0.1
¿Son estas probabilidades de que alguien esté vivo (es decir, status == 1) o de que alguien esté muerto (es decir, status == 0)?
Estoy bastante seguro de que son probabilidades de que alguien esté vivo, pero ¿es siempre así? ¿Hay alguna manera de especificar esto directamente en la función predict()?
davy.ai
Sí, las probabilidades devueltas por predict() corresponden a la posibilidad de que alguien esté vivo (es decir, estado == 1). Puede especificar esto directamente en la función predict() estableciendo el argumento type en “prob” en lugar de “response”. Esto devolverá un vector de probabilidades predichas de que el resultado sea 1 (vivo).