Предсказываем уход сотрудников!
В этом упражнении вы воспользуетесь созданными ранее переменными, чтобы предсказать, как долго нынешние сотрудники ещё останутся в компании.
В предыдущем упражнении вы обучили модель CoxPHFitter, сохранив её в объект cph. Также вы создали следующие переменные для прогнозирования:
current_employees: DataFrame со всеми сотрудниками, которые пока не уволились.current_employees_last_obs: Series, содержащий столбец длительностей из DataFramecurrent_employees.
Объект модели cph, DataFrame current_employees и Series current_employees_last_obs уже загружены. Теперь вы можете использовать их, чтобы помочь HR-отделу с задачей прогнозирования!
Это упражнение является частью курса
Анализ выживаемости в Python
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)