1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Phân tích sống sót bằng Python
Connected

Bài tập

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.

Hướng dẫn 1/2

undefined XP
    1
    2
  • Dự đoán thời gian trung vị đến lần bị bắt lại cho mỗi phạm nhân trong prison_new, gán kết quả dự đoán vào biến aft_pred.