KM 곡선으로 PH 가정 테스트하기
비례 위험 가정에 대해 배운 뒤, prison DataFrame으로 만든 Cox PH 모델을 아직 테스트하지 않았다는 것을 알게 되었어요. 이번에는 Kaplan-Meier 곡선을 사용해 범주형 변수 fin에 대해 비례 위험 가정을 검정해 보겠습니다.
cph는 모든 열을 사용해 적합된 Cox PH 모델이며, 이미 로드되어 있어요. KaplanMeierFitter 클래스의 인스턴스 kmf도 생성되어 있습니다.
CoxPHFitter 클래스, KaplanMeierFitter 클래스, 그리고 matplotlib.pyplot 모듈이 임포트되어 있어요. pandas와 numpy 라이브러리는 각각 pd, np로 임포트되어 있습니다.
이 연습은 강의의 일부입니다
Python에서의 Survival Analysis
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Fit to fin == 0
____
# Plot survival curve for fin == 0
ax = ____