Prever taxa de reencarceramento

Você desenvolveu um modelo de regressão de sobrevivência usando covariáveis sobre cada condenado libertado, incluindo estado civil, prisões anteriores etc. Dez novos condenados estão prestes a ser soltos, e você vai usar o modelo para prever o tempo mediano até a próxima prisão.

O novo DataFrame se chama prison_new e tem as mesmas colunas do DataFrame antigo, prison.

A classe WeibullAFTFitter foi importada e o modelo ajustado anteriormente foi carregado como aft. As bibliotecas pandas e numpy foram importadas como pd e np, respectivamente.