Förutsäg återhämtningsfrekvens
Du har utvecklat en överlevelseregressionsmodell med kovariater om varje frigiven fånge, inklusive civilstånd, tidigare häktningar och liknande. Tio nya fångar ska snart friges, och du ska använda modellen för att förutsäga deras mediantid till återhäktning.
Den nya DataFramen heter prison_new och har samma kolumner som den tidigare DataFramen prison.
Klassen WeibullAFTFitter har importerats och den tidigare tränade modellen har laddats som aft. Biblioteken pandas och numpy är importerade som pd respektive np.
Den här övningen är en del av kursen
Överlevnadsanalys i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Predict median of new data
aft_pred = ____