Wiederverhaftungsrate vorhersagen
Du hast ein Überlebensregressionsmodell entwickelt, das Kovariaten für jeden entlassenen Strafgefangenen enthält, darunter den Familienstand, frühere Verhaftungen usw. Zehn neue Strafgefangene stehen kurz vor ihrer Entlassung und du wirst das Modell nutzen, um ihre durchschnittliche Zeit bis zur erneuten Verhaftung vorherzusagen.
Der neue DataFrame heißt prison_new und hat die gleichen Spalten wie der alte DataFrame prison.
Die Klasse WeibullAFTFitter wurde importiert, und das zuvor angepasste Modell wurde als aft geladen. Die Bibliotheken pandas und numpy werden als pd bzw. np importiert.
Diese Übung ist Teil des Kurses
Überlebenszeitanalyse in Python
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Predict median of new data
aft_pred = ____