始める無料で始める

重要な特徴量を選ぶ

この演習では、最終モデルで使用する最も重要な特徴量だけを選択します。相対的重要度は、DataFrame relative_importances の列 importance に保存されていることを思い出してください。

この演習はコースの一部です

HRアナリティクス:Pythonで従業員離職を予測する

コースを見る

演習の手順

  • importance の値が 1% を超える特徴量のみを選択します。
  • それらの特徴量からリストを作成して出力します(この部分は用意されています)。
  • selected_list に保存されたインデックスを使って、features_trainfeatures_test の両方を、重要度が 1% を超える特徴量のみを含むように変換します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# select only features with relative importance higher than 1%
selected_features = relative_importances[relative_importances.____>0.01]

# create a list from those features: done
selected_list = selected_features.index

# transform both features_train and features_test components to include only selected features
features_train_selected = features_train[selected_list]
features_test_selected = ____[____]
コードを編集して実行