再逮捕率を予測する
既婚状況や前科件数など、出所者ごとの共変量を用いてサバイバル回帰モデルを作成しました。これから10人の新たな受刑者が出所予定で、モデルを使って彼らの再逮捕までの中央値(時間)を予測します。
新しいDataFrameはprison_newで、既存のDataFrame prisonと同じ列を持ちます。
WeibullAFTFitterクラスはインポート済みで、学習済みモデルはaftとして読み込まれています。pandasとnumpyはそれぞれpdとnpとしてインポートされています。
この演習はコースの一部です