1. 학습
  2. /
  3. 강의
  4. /
  5. Pythonで学ぶ木ベースのMachine Learning

Connected

연습 문제

特徴量重要度の可視化

この演習では、前の演習で学習したランダムフォレスト回帰器 rf による予測に、どの特徴量が最も効いていたかを確認します。

そのために、rf が評価した特徴量重要度の水平バー プロットを描画します。幸い、これは pandas の描画機能を使えば簡単に実行できます。

特徴量名を index、重要度を値に持つ pandas.Series オブジェクト importances を用意しています。さらに、matplotlib.pyplot は plt、pandas は pd として利用できます。

지침

100 XP
  • importances に対して .sort_values() を呼び出し、その結果を importances_sorted に代入します。

  • importances_sorted に対して .plot() を呼び出し、次の引数を設定します:

    • kind は 'barh'
    • color は 'lightgreen'