Testowanie założenia PH za pomocą krzywych KM
Po zapoznaniu się z założeniem proporcjonalności hazardów zauważasz, że zbudowany przez ciebie model Cox PH na podstawie ramki danych prison nie został jeszcze przetestowany. Przetestujesz to założenie dla zmiennej kategorycznej fin, korzystając z krzywych Kaplana-Meiera.
cph to dopasowany model Cox PH z użyciem wszystkich kolumn – jest już załadowany. Klasa KaplanMeierFitter została zainicjalizowana jako kmf.
Klasy CoxPHFitter i KaplanMeierFitter oraz moduł matplotlib.pyplot zostały zaimportowane. Biblioteki pandas i numpy są dostępne odpowiednio jako pd i np.
To ćwiczenie jest częścią kursu
Analiza przeżycia w Pythonie
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____