1. Learn
  2. /
  3. Courses
  4. /
  5. HRアナリティクス:Pythonで従業員離職を予測する

Connected

Exercise

従業員離職モデルの比較

この演習では、剪定済みツリー(max_depth=7)を使って、バランスあり モデルと バランスなし(デフォルト)モデルを比較します。バランスなし モデルについては、すでに recall と ROC/AUC スコアを計算済みです。バランスあり モデルでも同じ手順を完成させてください。

  • 変数 features_train、target_train、features_test、target_test はワークスペースに用意されています。
  • バランスなしモデルはすでに学習済みで、その予測は prediction に保存されています。
  • recall_score() と roc_auc_score() 関数はインポート済みです。

Instructions

100 XP
  • バランスありモデルを初期化し、最大深さを 7、乱数シードを 42 に設定します。
  • トレーニングセットを使って学習用データにフィットさせます。
  • テストセットを使って予測を行います。
  • recall スコア と ROC/AUC スコアを出力します。