¡Predice antes de que se vayan!
En este ejercicio, vas a usar las variables que acabas de crear para predecir cuánto tiempo más se quedarán las personas empleadas actualmente.
En el ejercicio anterior, ajustaste los datos a un objeto modelo CoxPHFitter llamado cph. También creaste las siguientes variables para usarlas en la predicción:
current_employees: DataFrame con todas las personas empleadas que siguen en la empresa.current_employees_last_obs: Series que hace referencia a la columna de duraciones del DataFramecurrent_employees.
El objeto de modelo cph al que acabas de ajustar los datos, el DataFrame current_employees y la Series current_employees_last_obs están cargados para ti. ¡Ahora puedes usarlos para ayudar al departamento de RR. HH. con esta tarea de predicción!
Este ejercicio forma parte del curso
Análisis de supervivencia en Python
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)