CommencezCommencez gratuitement

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".

Cette activité fait partie du cours

Apprentissage supervisé avec scikit-learn

Voir le cours

Instructions de l’exercice

  • 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.

Exercice interactif pratique

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

# Create music_dummies
music_dummies = ____

# Print the new DataFrame's shape
print("Shape of music_dummies: {}".format(____))
Modifier et exécuter le code