Prevedi il tasso di nuova arresto
Hai sviluppato un modello di regressione di sopravvivenza usando covariate per ogni detenuto rilasciato, inclusi stato civile, arresti precedenti, ecc. Dieci nuovi detenuti stanno per essere rilasciati e userai il modello per prevedere il tempo mediano al nuovo arresto.
Il nuovo DataFrame si chiama prison_new e ha le stesse colonne del vecchio DataFrame prison.
La classe WeibullAFTFitter è stata importata e il modello precedentemente addestrato è stato caricato come aft. Le librerie pandas e numpy sono importate rispettivamente come pd e np.
Questo esercizio fa parte del corso
Analisi di sopravvivenza in Python
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Predict median of new data
aft_pred = ____