1. 학습
    2. /
  2. 강의
    3. /
  3. Python에서의 Survival Analysis
Connected

연습 문제

재구금률 예측

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

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

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

지침 1/2

undefined XP
    1
    2
  • prison_new의 각 출소자에 대해 재구금까지의 중앙값 시간을 예측하고, 예측값을 aft_pred 변수에 할당하세요.