Prédire le taux de ré‑arrestation

Vous avez développé un modèle de régression de survie en utilisant des covariables pour chaque détenu libéré, notamment leur statut matrimonial, leurs arrestations antérieures, etc. Dix nouveaux détenus vont être libérés, et vous allez utiliser 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 ajusté précédemment a été chargé sous le nom aft. Les bibliothèques pandas et numpy sont importées sous les alias pd et np, respectivement.