재구금률 예측

혼인 여부, 이전 체포 이력 등 각 출소자의 공변량을 사용해 생존 회귀 모델을 개발했어요. 이제 곧 10명의 출소자가 나오며, 이 모델을 사용해 재구금까지의 중앙값 시간을 예측하려고 합니다.

새로운 DataFrame은 prison_new이며, 기존 DataFrame prison과 동일한 열을 갖습니다.

WeibullAFTFitter 클래스는 임포트되어 있고, 이전에 학습된 모델은 aft로 로드되어 있습니다. pandas와 numpy 라이브러리는 각각 pd, np로 임포트되어 있습니다.