Sage es voraus, bevor sie gehen!
In dieser Aufgabe verwendest du die Variablen, die du gerade erstellt hast, um vorherzusagen, wie lange die aktuellen Mitarbeiter bleiben werden.
In der vorherigen Aufgabe hast du die Daten an ein CoxPHFitter
-Modellobjekt namens cph
angepasst. Du hast auch die folgenden Variablen für die Vorhersage erstellt:
current_employees
: DataFrame für alle Mitarbeiter, die noch da sind.current_employees_last_obs
: Series, die sich auf die durations-Spalte des DataFramecurrent_employees
bezieht.
Das Modellobjekt cph
, an das du die Daten gerade angepasst hast, DataFrame current_employees
und Series current_employees_last_obs
werden für dich geladen. Jetzt kannst du sie benutzen, um der HR-Abteilung bei dieser Vorhersageaufgabe zu helfen!
Diese Übung ist Teil des Kurses
Überlebenszeitanalyse in Python
Interaktive Übung zum Anfassen
Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)