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

Exercice

Pipeline pour la prédiction du genre musical : II

Après avoir défini les étapes du pipeline dans l'exercice précédent, vous allez maintenant l'utiliser sur l'ensemble de données music_df pour classifier le genre des chansons. Ce qui rend les pipelines si utiles, c'est l'interface simple qu'ils offrent.

X_train, X_test, y_train et y_test ont été préchargés pour vous, et confusion_matrix a été importée de sklearn.metrics.

Instructions

100 XP
  • Créez un pipeline en utilisant les étapes que vous avez déjà définies.
  • Ajustez le pipeline sur les données d'entraînement.
  • Faites des prédictions sur l'ensemble de test.
  • Calculez et affichez la matrice de confusion.