1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Phân tích sống sót bằng Python

Connected

Bài tập

Nghiên cứu nghỉ việc của nhân viên

Tuyển mới để thay thế nhân viên nghỉ việc sẽ phát sinh chi phí tuyển dụng và đào tạo. Bạn muốn dự đoán nhân viên hiện tại sẽ gắn bó bao lâu. Bài tập này tập trung vào các bước đầu tiên để chuẩn bị cho việc dự đoán.

Bạn có một DataFrame tên là employees. Nó chứa dữ liệu về 1470 nhân viên (đã nghỉ và còn làm) cùng câu trả lời khảo sát của họ. Khảo sát bao gồm các khía cạnh sau:

  • environment_satisfaction
  • job_satisfaction
  • relationship_satisfaction
  • work_life_balance

Ngoài ra, years_at_company là thời gian nhân viên đã làm việc và attrition cho biết nhân viên đã nghỉ việc hay chưa (1 nếu đã nghỉ, 0 nếu chưa).

Một vài dòng mẫu đã được in sẵn trên console. Lớp CoxPHFitter đã được import từ gói lifelines.

Hướng dẫn 1/2

undefined XP
    1
    2
  • Khởi tạo một đối tượng CoxPHFitter.
  • Fit mô hình CoxPHFitter với dữ liệu của bạn, sử dụng tất cả các cột còn lại.