Predicția ratei de rearestare
Ai dezvoltat un model de regresie de supraviețuire folosind covariate despre fiecare deținut eliberat, inclusiv statutul marital, arestările anterioare etc. Zece deținuți noi urmează să fie eliberați, iar tu vei folosi modelul pentru a prezice timpul median până la rearestare.
Noul DataFrame se numește prison_new și are aceleași coloane ca și DataFrame-ul anterior prison.
Clasa WeibullAFTFitter a fost importată, iar modelul antrenat anterior a fost încărcat ca aft. Bibliotecile pandas și numpy sunt importate ca pd și, respectiv, np.
Acest exercițiu face parte din cursul
Analiza de supraviețuire în Python
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Predict median of new data
aft_pred = ____