始める無料で始める

再逮捕率を予測する

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

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

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

この演習はコースの一部です

Pythonで学ぶSurvival Analysis

コースを見る

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Predict median of new data
aft_pred = ____
コードを編集して実行