Dự đoán trước khi họ nghỉ việc!
Trong bài tập này, bạn sẽ dùng các biến vừa tạo để dự đoán những nhân viên hiện tại sẽ ở lại bao lâu.
Ở bài trước, bạn đã fit dữ liệu vào một đối tượng mô hình CoxPHFitter tên là cph. Bạn cũng đã tạo các biến sau để dùng cho việc dự đoán:
current_employees: DataFrame chứa tất cả nhân viên vẫn đang làm việc.current_employees_last_obs: Series tham chiếu đến cột thời lượng (durations) của DataFramecurrent_employees.
Đối tượng mô hình cph bạn vừa fit dữ liệu, DataFrame current_employees, và Series current_employees_last_obs đã được nạp sẵn cho bạn. Giờ hãy dùng chúng để hỗ trợ bộ phận HR trong nhiệm vụ dự đoán này!
Bài tập này là một phần của khóa học
Phân tích sống sót bằng Python
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)