KM曲線でPH仮定を検定する
比例ハザードの仮定について学んだ結果、prison データフレームで作成したCox PHモデルがまだ検定されていないことに気づきました。ここでは、カテゴリ変数 fin に対してKaplan-Meier曲線を用いて比例ハザードの仮定を検定します。
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 = ____