Спрогнозуйте рівень повторного арешту

Ви розробили модель регресії виживання з ковараіатами для кожного звільненого засудженого, зокрема сімейний стан, попередні арешти тощо. Невдовзі звільнять десятьох нових засуджених, і ви використаєте модель, щоб передбачити їхній медіанний час до повторного арешту.

Новий датафрейм має назву prison_new і містить ті самі стовпці, що й старий датафрейм prison.

Клас WeibullAFTFitter імпортовано, а раніше навчена модель завантажена як aft. Бібліотеки pandas та numpy імпортовано як pd і np відповідно.