Vizualizace vlivu kovariát na přežití
Dva vězni čekají na propuštění. Chceš pomocí modelu CoxPHFitter, který jsi právě natrénoval/a, předpovědět jejich funkci přežití pro dobu do opětovného zatčení. Hodnoty kovariát pro jednotlivé osoby jsou uvedeny níže:
| Jméno | fin | age | wexp | mar | paro | prio |
|---|---|---|---|---|---|---|
| Jack | 0 | 35 | 0 | 1 | 1 | 3 |
| Marie | 1 | 22 | 0 | 0 | 0 | 0 |
Pomocí cph vykresli křivky přežití Jacka a Marie vedle sebe spolu s výchozí křivkou přežití. Natrénovaný model cph je již načtený.
Třída CoxPHFitter je importovaná a modul matplotlib.pyplot je importovaný jako plt. Knihovny pandas a numpy jsou importované jako pd a np.
Toto cvičení je součástí kurzu
Survival Analysis in Python
Pokyny k cvičení
- Pomocí
.plot_partial_effects_on_outcome()zadej hodnoty kovariát (uvedené výše) pro Jacka a Marii a vykresli jejich křivky přežití. - Zobraz graf.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Plot partial effects
cph.____(covariates=____,
values=____)
# Show plot
plt.show()