Prueba el supuesto de PH con curvas KM
Tras aprender sobre el supuesto de riesgos proporcionales, te das cuenta de que el modelo de Cox PH que construiste con el DataFrame prison no se ha puesto a prueba. Vas a comprobar el supuesto de riesgos proporcionales para la variable categórica fin usando curvas de Kaplan-Meier.
cph es el modelo de Cox PH ajustado con todas las columnas y ya está cargado. También se ha instanciado una clase KaplanMeierFitter llamada kmf.
Se han importado la clase CoxPHFitter, la clase KaplanMeierFitter y el módulo matplotlib.pyplot. Las librerías pandas y numpy están importadas como pd y np, respectivamente.
Este ejercicio forma parte del curso
Análisis de supervivencia en Python
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____