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 zum Anfassen
Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.
# Predict median of new data
aft_pred = ____