Förutsäg innan de slutar!
I den här övningen använder du variablerna du just skapade för att förutsäga hur länge nuvarande anställda kommer att stanna.
I föregående övning anpassade du data till ett CoxPHFitter-modellobjekt kallat cph. Du skapade även följande variabler för prediktion:
current_employees: DataFrame för alla anställda som fortfarande är kvar.current_employees_last_obs: Series som refererar till varaktighetskolumnen i DataFramecurrent_employees.
Modellobjektet cph som du anpassade data till, DataFrame current_employees och Series current_employees_last_obs är inlästa åt dig. Nu kan du använda dem för att hjälpa HR-avdelningen med den här prediktionsuppgiften!
Den här övningen är en del av kursen
Överlevnadsanalys i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Predict survival function conditional on existing durations
____.predict_survival_function(____)