Передбачте, перш ніж вони підуть!
У цій вправі ви використаєте щойно створені змінні, щоб передбачити, як довго залишаться нинішні співробітники.
У попередній вправі ви натренували дані на об'єкті моделі CoxPHFitter під назвою cph. Ви також створили такі змінні для прогнозування:
current_employees: датафрейм для всіх співробітників, які все ще працюють у компанії.current_employees_last_obs: серія, що відповідає стовпцю тривалостей у датафрейміcurrent_employees.
Об'єкт моделі cph, на якому ви щойно потренували дані, а також датафрейм current_employees і серія current_employees_last_obs уже завантажені для вас. Тепер ви можете використати їх, щоб допомогти відділу кадрів із цим завданням прогнозування!
Ця вправа є частиною курсу
Аналіз виживаності в Python
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)