1. 学ぶ
    2. /
  2. コース
    3. /
  3. scikit-learn で学ぶ Supervised Learning
Connected

演習

楽曲ジャンル予測のパイプライン：II

前の演習でパイプラインのステップを設定したので、今回はそれを music_df データセットに適用し、楽曲のジャンルを分類します。パイプラインが非常に有用なのは、シンプルなインターフェースを提供してくれる点です。

X_train、X_test、y_train、y_test はあらかじめ用意されており、confusion_matrix は sklearn.metrics からインポート済みです。

指示

100 XP
  • 先ほど定義したステップを使ってパイプラインを作成します。
  • パイプラインを学習データに適合させます。
  • テストセットで予測を行います。
  • 混同行列を計算して表示します。