Prediksi tingkat penangkapan kembali

Anda telah mengembangkan model regresi survival menggunakan kovariat tentang setiap narapidana yang dibebaskan, termasuk status perkawinan, riwayat penangkapan, dan lain-lain. Sepuluh narapidana baru akan segera dibebaskan, dan Anda akan menggunakan model tersebut untuk memprediksi median waktu hingga mereka ditangkap kembali.

DataFrame baru bernama prison_new dan memiliki kolom yang sama dengan DataFrame lama prison.

Kelas WeibullAFTFitter telah diimpor, dan model yang telah dilatih sebelumnya dimuat sebagai aft. Pustaka pandas dan numpy masing-masing diimpor sebagai pd dan np.