Prédire le taux de ré-arrestation
Vous avez élaboré un modèle de régression de survie en utilisant des covariables pour chaque détenu libéré, notamment l'état civil, les arrestations antérieures, etc. Dix nouveaux détenus sont sur le point d'être libérés, et vous utiliserez le modèle pour prédire leur temps médian avant une nouvelle arrestation.
Le nouveau DataFrame s'appelle prison_new et possède les mêmes colonnes que l'ancien DataFrame prison.
La classe WeibullAFTFitter a été importée, et le modèle précédemment ajusté a été chargé sous le nom aft. Les bibliothèques pandas et numpy sont importées respectivement sous pd et np.
Cette activité fait partie du cours
Analyse de survie en Python
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Predict median of new data
aft_pred = ____