Перевірте припущення про пропорційні ризики за допомогою кривих Каплана—Маєра
Після ознайомлення з припущенням про пропорційні ризики ви розумієте, що побудована вами модель Cox PH на основі датафрейму prison не була перевірена. Ви перевірите припущення про пропорційні ризики для категоріальної змінної fin за допомогою кривих Каплана—Маєра.
cph — це навчена модель Cox PH з використанням усіх стовпців, її вже завантажено. Також створено екземпляр класу KaplanMeierFitter під назвою kmf.
Класи CoxPHFitter, KaplanMeierFitter та модуль matplotlib.pyplot імпортовано. Бібліотеки pandas та numpy імпортовано як pd і np відповідно.
Ця вправа є частиною курсу
Аналіз виживаності в Python
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____