1. Apprendre
  2. /
  3. Cours
  4. /
  5. Apprentissage supervisé avec scikit-learn

Connected

Exercice

Créer des variables fictives

Pouvoir inclure des caractéristiques catégorielles dans la construction du modèle peut améliorer la performance, puisqu'elles peuvent ajouter de l'information utile à la justesse des prédictions.

Le jeu de données music_df a déjà été chargé pour vous et sa forme a été affichée. De plus, pandas a été importée sous le nom pd.

Vous allez maintenant créer un nouveau DataFrame qui contient les colonnes originales de music_df ainsi que des variables fictives dérivées de la colonne "genre".

Instructions

100 XP
  • Utilisez une fonction pertinente, en passant le DataFrame complet music_df, pour créer music_dummies, en supprimant la première colonne binaire.
  • Affichez la forme de music_dummies.