1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Analiza de supraviețuire în Python
Connected

Bài tập

Reprezentarea grafică a efectelor covarianților asupra supraviețuirii

Doi deținuți se află pe lista de așteptare pentru liberare. Vrei să prezici funcția de supraviețuire pentru timpul până la rearestare, folosind modelul CoxPHFitter pe care tocmai l-ai antrenat. Valorile covarianților pentru cei doi indivizi sunt listate mai jos:

Nume fin age wexp mar paro prio
Jack 0 35 0 1 1 3
Marie 1 22 0 0 0 0

Reprezintă grafic curbele de supraviețuire ale lui Jack și Marie, alături de curba de supraviețuire de referință, folosind cph. Modelul antrenat cph a fost deja încărcat pentru tine.

Clasa CoxPHFitter a fost importată, iar modulul matplotlib.pyplot a fost importat ca plt. Bibliotecile pandas și numpy sunt importate ca pd, respectiv np.

Hướng dẫn

100 XP
  • Folosește .plot_partial_effects_on_outcome() pentru a specifica valorile covarianților (listate mai sus) pentru Jack și Marie și pentru a reprezenta grafic curbele lor de supraviețuire.
  • Afișează graficul.