1. Apprendre
    2. /
  2. Cours
    3. /
  3. Modèles linéaires généralisés en Python
Connected

Exercice

Modéliser avec une variable catégorielle

Dans les exercices précédents, vous avez ajusté un modèle de régression logistique avec color comme variable explicative en plus de width, en traitant color comme une variable quantitative. Dans cet exercice, vous allez plutôt traiter color comme une variable catégorielle. Ainsi, lors de la construction de la matrice du modèle, color sera encodée en 3 variables avec un codage 0/1.

Rappelez-vous que l'encodage par défaut dans dmatrix() utilise le premier groupe comme groupe de référence. Pour afficher la matrice du modèle sous forme de dataframe, un argument additionnel de dmatrix(), soit return_type, sera défini à 'dataframe'.

La variable color possède un ordre naturel comme suit :
1 : medium light
2 : medium
3 : medium dark
4 : dark

Le jeu de données crab est déjà chargé dans l'espace de travail.

Instructions 1/4

undefined XP
    1
    2
    3
    4
  • Construisez une model_matrix avec color comme variable. color doit être traitée comme catégorielle et le groupe de référence doit être fixé à 4 à l'aide de la fonction Treatment().
  • Ajustez un modèle logistique avec y comme variable réponse et affichez les résultats.