1. Learn
  2. /
  3. Courses
  4. /
  5. Survival Analysis in Python

Connected

Exercise

Predict re-arrest rate

You developed a survival regression model using covariates about each released convict, including their marital status, prior arrests, etc. Ten new convicts are about to be released, and you will use the model to predict their median time to re-arrest.

The new DataFrame is called prison_new and has the same columns as the old DataFrame prison.

The WeibullAFTFitter class has been imported, and the previously fitted model has been loaded as aft. The pandas and numpy libraries are imported as pd and np, respectively.

Instructions 1/2

undefined XP
    1
    2
  • Predict the median time to arrest for each convict in prison_new, assign the predictions to a variable named aft_pred.