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
Instructions de l’exercice
- Utilisez une fonction pertinente, en passant le DataFrame complet
music_df, pour créermusic_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(____))