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

Kurs anzeigen

Interaktive Übung zum Anfassen

Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.

# Predict median of new data
aft_pred = ____