1. 学ぶ
    2. /
  2. コース
    3. /
  3. HRアナリティクス：Pythonで従業員離職を予測する
Connected

演習

木のエクスポート

Decision Tree の分類タスクでは、木が深くなりすぎると過学習が起きやすくなります。学習データとテストデータの精度を比較すると、結果に過学習があることが分かります。これは木の可視化からも読み取れます。

この演習では、決定木をテキストドキュメントにエクスポートし、後で可視化に利用できるようにします。

指示

100 XP
  • sklearn.tree サブモジュールから export_graphviz() 関数をインポートします。
  • model を学習データにフィットさせます。
  • 可視化を tree.dot というファイルにエクスポートします。