Teste a suposição de PH com curvas de KM
Depois de aprender sobre a suposição de riscos proporcionais, você percebe que o modelo de Cox PH que construiu usando o DataFrame prison não foi testado. Você vai testar a suposição de riscos proporcionais para a variável categórica fin usando curvas de Kaplan-Meier.
cph é o modelo de Cox PH ajustado usando todas as colunas e já foi carregado. A classe KaplanMeierFitter kmf também foi instanciada.
As classes CoxPHFitter, KaplanMeierFitter e o módulo matplotlib.pyplot foram importados. As bibliotecas pandas e numpy foram importadas como pd e np, respectivamente.
Este exercício faz parte do curso
Análise de Sobrevivência em Python
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____