1. เรียนรู้
  2. /
  3. Courses
  4. /
  5. Pythonで学ぶSurvival Analysis

Connected

Exercises

再逮捕率を予測する

既婚状況や前科件数など、出所者ごとの共変量を用いてサバイバル回帰モデルを作成しました。これから10人の新たな受刑者が出所予定で、モデルを使って彼らの再逮捕までの中央値(時間)を予測します。

新しいDataFrameはprison_newで、既存のDataFrame prisonと同じ列を持ちます。

WeibullAFTFitterクラスはインポート済みで、学習済みモデルはaftとして読み込まれています。pandasとnumpyはそれぞれpdとnpとしてインポートされています。

คำแนะนำ 1 / 2

undefined XP
    1
    2
  • prison_newの各受刑者について再逮捕までの中央値を予測し、変数aft_predに代入します。