Get startedGet started for free

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.

This exercise is part of the course

Survival Analysis in Python

View Course

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# Predict median of new data
aft_pred = ____
Edit and Run Code