Cox PH で刑務所データをモデリングする
DataFrame prison には、釈放後1年間追跡された432名の受刑者に関する情報が含まれています。これまでに Weibull AFT モデルを使って逮捕までの時間をモデル化し、どの要因が再逮捕のリスクを高めたり低下させたりするかを調べました。
lifelines の CoxPHFitter クラスは、生存回帰のための Cox 比例ハザードモデルを実装しており、ベースラインハザード関数と、ハザードの比率(ハザード比)をモデル化します。CoxPHFitter を使って、要因を探ってみましょう!
pandas と numpy はそれぞれ pd と np としてインポート済みです。必要に応じてコンソールで DataFrame とその列名を確認してください。
この演習はコースの一部です
Pythonで学ぶSurvival Analysis
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Import CoxPHFitter class
____
# Instantiate CoxPHFitter class cph
____