Проверка допущения о пропорциональных рисках с помощью кривых KM
Изучив допущение о пропорциональных рисках, вы понимаете, что построенная модель Cox PH на основе DataFrame 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 = ____