CommencerCommencez gratuitement

Prédire la survie des moineaux

Dans cet exercice, vous allez prédire la probabilité de survie en utilisant le modèle de survie des moineaux de l’exercice précédent.

Rappelez-vous que lors de l’appel à predict() (docs) pour obtenir les probabilités prédites à partir d’un modèle glm(), vous devez préciser que vous souhaitez la réponse :

predict(model, type = "response")

Sinon, predict() sur un modèle de régression logistique renvoie les log-cotes (log-odds) prédites de l’événement, et non la probabilité.

Vous utiliserez aussi la fonction GainCurvePlot() (docs) pour tracer la courbe de gain à partir des prédictions du modèle. Si la courbe de gain du modèle est proche de la courbe de gain idéale ("wizard"), alors le modèle trie bien les moineaux : autrement dit, il prédit une probabilité de survie plus élevée pour les moineaux qui ont effectivement survécu. Les entrées de la fonction GainCurvePlot() sont :

  • frame : data frame avec la colonne de prédiction et la colonne de vérité terrain
  • xvar : le nom de la colonne des prédictions (en chaîne de caractères)
  • truthVar : le nom de la colonne avec le résultat observé (en chaîne de caractères)
  • title : un titre pour le graphique (en chaîne de caractères)

GainCurvePlot(frame, xvar, truthVar, title)

Le data frame sparrow et le modèle sparrow_model ont été préchargés.

Cet exercice fait partie du cours

<cours>Apprentissage supervisé en R : Régression</cours>
Voir le cours

Instructions de l’exercice

  • Créez une nouvelle colonne dans sparrow appelée pred qui contient les prédictions sur les données d’entraînement.
  • Appelez GainCurvePlot() pour créer la courbe de gain des prédictions. Le modèle parvient-il à bien trier les moineaux selon qu’ils ont effectivement survécu ou non ?

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# sparrow is available
summary(sparrow)

# sparrow_model is available
summary(sparrow_model)

# Make predictions
sparrow$pred <- ___

# Look at gain curve
___(___, ___, ___, "sparrow survival model")
Modifier et exécuter le code