Testa PH-antagandet med KM-kurvor
Efter att ha lärt dig om antagandet om proportionella hazarder inser du att Cox PH-modellen du byggde med prison-DataFramen inte har testats. Du ska testa antagandet om proportionella hazarder för den kategoriska variabeln fin med hjälp av Kaplan-Meier-kurvor.
cph är den anpassade Cox PH-modellen som använder alla kolumner och har redan laddats in. En KaplanMeierFitter-klass kmf har också instansierats.
Klasserna CoxPHFitter och KaplanMeierFitter samt modulen matplotlib.pyplot har importerats. Biblioteken pandas och numpy är importerade som pd respektive np.
Den här övningen är en del av kursen
Överlevnadsanalys i Python
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____