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

演習

従業員データへの決定木の適合

学習用とテスト用に分割すると、学習データで分類器を作成し、残りのデータで評価できます。この演習では、decision tree 分類アルゴリズムを使って従業員の離職予測モデルの作成を始めます。アルゴリズムは .fit() メソッドを提供しており、学習用データの特徴量をモデルに当てはめることができます。

【リマインダー】目的変数と特徴量はすでに学習用とテスト用に分割済みです（Train: features_train, target_train、Test: features_test, target_test）。

指示

100 XP
  • DecisionTreeClassifier という分類アルゴリズムをインポートします。
  • それを model として初期化し、random_state を 42 に設定します。
  • 学習用の特徴量を model に当てはめて、決定木モデルを適用します。