Predecir la tasa de re-arresto

Has desarrollado un modelo de regresión de supervivencia usando covariables de cada convicto puesto en libertad, como su estado civil, arrestos previos, etc. Diez nuevos convictos están a punto de ser liberados y vas a usar el modelo para predecir su tiempo mediano hasta el re-arresto.

El nuevo DataFrame se llama prison_new y tiene las mismas columnas que el DataFrame anterior prison.

La clase WeibullAFTFitter ha sido importada y el modelo ajustado previamente se ha cargado como aft. Las librerías pandas y numpy están importadas como pd y np, respectivamente.