Прогнозирование повторных арестов

Вы разработали модель регрессии выживаемости, используя ковариаты для каждого освобождённого осуждённого: семейное положение, количество предыдущих арестов и другие характеристики. Десять новых осуждённых готовятся к освобождению, и вы воспользуетесь моделью, чтобы предсказать медианное время до повторного ареста.

Новый DataFrame называется prison_new и содержит те же столбцы, что и исходный DataFrame prison.

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