他们何时离职?先做个预测!
在本练习中,您将使用刚刚创建的变量来预测在职员工还会待多久。
在上一个练习中,您已将数据拟合到名为 cph 的 CoxPHFitter 模型对象。您还创建了以下用于预测的变量:
current_employees:包含所有仍在职员工的 DataFrame。current_employees_last_obs:指向 DataFramecurrent_employees的持续时间列的 Series。
我们已为您加载了刚刚拟合好的模型对象 cph、DataFrame current_employees,以及 Series current_employees_last_obs。现在就用它们来帮助人力资源部门完成这项预测任务吧!
本练习是课程的一部分
Python 中的生存分析
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Predict survival function conditional on existing durations
____.predict_survival_function(____)