Да здравствует демократия!
Ранее в этой главе мы изучили и подтвердили проблему цензурирования данных в наборе данных о продолжительности режимов. Теперь пришло время построить первую кривую выживаемости для оценки средней продолжительности.
Вам нужно оценить функцию выживаемости и построить кривую выживаемости, отображающую вероятности выживания. Для оценки функции выживаемости создайте экземпляр объекта KaplanMeierFitter и обучите его на соответствующих столбцах данных. Имена параметров уже указаны — подробнее о подгонке функций выживаемости вы узнаете в следующих главах.
Пакет pandas загружен как pd, модуль matplotlib.pyplot — как plt, а класс KaplanMeierFitter импортирован из lifelines. DataFrame с продолжительностями загружен и сохранён как regime_durations.
Это упражнение является частью курса
Анализ выживаемости в Python
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Instantiate a KaplanMeierFitter object kmf
kmf = ____