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.
Acest exercițiu face parte din cursul
Analiza de supraviețuire în Python
Instrucțiuni pentru exercițiu
- 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.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Plot partial effects
cph.____(covariates=____,
values=____)
# Show plot
plt.show()