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 = ____