1. 学ぶ
    2. /
  2. コース
    3. /
  3. Pythonで学ぶSurvival Analysis
Connected

演習

従業員の離職分析

欠員補充のための新規採用には、採用コストや研修コストがかかります。そこで、現在在籍している従業員がどのくらい勤務を続けるかを予測したいと考えています。この演習では、予測の準備として最初のステップに取り組みます。

employees という DataFrame が与えられています。これは、1470 名の従業員（離職者と在籍者）について、調査への回答を含むデータです。調査は次の指標で構成されています。

  • environment_satisfaction
  • job_satisfaction
  • relationship_satisfaction
  • work_life_balance

さらに、years_at_company は従業員の在籍年数を表し、attrition は従業員が離職したかどうかを示します（離職なら 1、そうでなければ 0）。

サンプル行はコンソールに表示されています。lifelines パッケージから CoxPHFitter クラスはすでにインポートされています。

指示1 / 2

undefined XP
    1
    2
  • CoxPHFitter オブジェクトをインスタンス化します。
  • すべての他の列を用いて、CoxPHFitter のモデルオブジェクトをデータで学習させます。