Rappresenta gli effetti delle covariate sulla sopravvivenza
Due detenuti sono in lista d'attesa per il rilascio. Vuoi prevedere la loro funzione di sopravvivenza per il tempo all'arresto usando il modello CoxPHFitter che hai appena adattato. I valori delle covariate per gli individui sono elencati di seguito:
| Nome | fin | age | wexp | mar | paro | prio |
|---|---|---|---|---|---|---|
| Jack | 0 | 35 | 0 | 1 | 1 | 3 |
| Marie | 1 | 22 | 0 | 0 | 0 | 0 |
Rappresenta affiancate le curve di sopravvivenza di Jack e Marie insieme alla curva di sopravvivenza di base usando cph. Il modello adattato cph è già stato caricato per te.
La classe CoxPHFitter è stata importata e il modulo matplotlib.pyplot è stato importato come plt. Le librerie pandas e numpy sono importate rispettivamente come pd e np.
Questo esercizio fa parte del corso
Analisi di sopravvivenza in Python
Istruzioni dell'esercizio
- Usa
.plot_partial_effects_on_outcome()per specificare i valori delle covariate (elencati sopra) per Jack e Marie e traccia le loro curve di sopravvivenza. - Mostra il grafico.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Plot partial effects
cph.____(covariates=____,
values=____)
# Show plot
plt.show()