1. Learn
  2. /
  3. Courses
  4. /
  5. Analiza de supraviețuire în Python

Connected

Exercise

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.

Instructions 1/2

undefined XP
    1
    2
  • Prezice timpul median până la arest pentru fiecare deținut din prison_new și atribuie predicțiile unei variabile numite aft_pred.