Hay dos reclusos en lista de espera para ser liberados. Quieres predecir su función de supervivencia para el tiempo hasta el arresto usando el modelo CoxPHFitter que acabas de ajustar. Los valores de las covariables para cada individuo se muestran a continuación:

Name	fin	age	wexp	mar	paro	prio
Jack	0	35	0	1	1	3
Marie	1	22	0	0	0	0

Representa, junto a la curva de supervivencia de referencia, las curvas de supervivencia de Jack y Marie usando cph. El modelo ajustado cph ya está cargado.

La clase CoxPHFitter se ha importado y el módulo matplotlib.pyplot se ha importado como plt. Las librerías pandas y numpy se han importado como pd y np, respectivamente.

Predecir con el modelo de Cox PH

Create Your Free Account