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.
Deze oefening maakt deel uit van de cursus
Survival Analysis in Python
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Predict median of new data
aft_pred = ____