Prédire avant qu’ils partent !

Dans cet exercice, vous allez utiliser les variables que vous venez de créer pour prédire combien de temps les employés actuels vont rester.

Dans l’exercice précédent, vous avez ajusté les données à un objet modèle CoxPHFitter appelé cph. Vous avez également créé les variables suivantes à utiliser pour la prédiction :

• current_employees : DataFrame pour tous les employés encore en poste.
• current_employees_last_obs : Series correspondant à la colonne des durées du DataFrame current_employees.

L’objet modèle cph auquel vous venez d’ajuster les données, le DataFrame current_employees et la Series current_employees_last_obs sont déjà chargés. Vous pouvez maintenant les utiliser pour aider le service RH dans cette tâche de prédiction !