Voorspel het herarresteringspercentage
Je hebt een survivalregressiemodel ontwikkeld met covariabelen over elke vrijgelaten gevangene, zoals hun burgerlijke staat, eerdere arrestaties, enzovoort. Tien nieuwe gevangenen staan op het punt vrijgelaten te worden, en je gaat het model gebruiken om hun mediane tijd tot herarrest te voorspellen.
De nieuwe DataFrame heet prison_new en heeft dezelfde kolommen als de oude DataFrame prison.
De klasse WeibullAFTFitter is geïmporteerd en het eerder getrainde model is geladen als aft. De bibliotheken pandas en numpy zijn respectievelijk geïmporteerd als pd en np.
Deze oefening maakt deel uit van de cursus
Survivalanalyse in Python
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Predict median of new data
aft_pred = ____