Dự đoán tỷ lệ bị bắt lại

Bạn đã xây dựng một mô hình hồi quy sống sót sử dụng các biến đồng biến liên quan đến từng phạm nhân được thả, bao gồm tình trạng hôn nhân, số lần bị bắt trước đó, v.v. Mười phạm nhân mới sắp được thả, và bạn sẽ dùng mô hình để dự đoán thời gian trung vị đến lần bị bắt lại.

DataFrame mới tên là prison_new và có cùng các cột như DataFrame cũ prison.

Lớp WeibullAFTFitter đã được nhập, và mô hình đã huấn luyện trước đó được nạp dưới tên aft. Thư viện pandas và numpy đã được nhập tương ứng là pd và np.