Vẽ ảnh hưởng của biến đồng biến lên sống sót
Hai phạm nhân đang trong danh sách chờ được thả. Bạn muốn dự đoán hàm sống sót cho thời gian đến khi bị bắt lại bằng mô hình CoxPHFitter mà bạn vừa fit. Giá trị các biến đồng biến cho từng người như sau:
| Name | fin | age | wexp | mar | paro | prio |
|---|---|---|---|---|---|---|
| Jack | 0 | 35 | 0 | 1 | 1 | 3 |
| Marie | 1 | 22 | 0 | 0 | 0 | 0 |
Vẽ đường cong sống sót của Jack và Marie cạnh nhau cùng với đường cong sống sót cơ sở bằng cph. Mô hình đã fit cph đã được nạp sẵn cho bạn.
Lớp CoxPHFitter đã được import và mô-đun matplotlib.pyplot đã được import dưới tên plt. Thư viện pandas và numpy lần lượt được import dưới tên pd và np.
Bài tập này là một phần của khóa học
Phân tích sống sót bằng Python
Hướng dẫn bài tập
- Dùng
.plot_partial_effects_on_outcome()để chỉ định các giá trị biến đồng biến (liệt kê ở trên) cho Jack và Marie và vẽ các đường cong sống sót của họ. - Hiển thị biểu đồ.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Plot partial effects
cph.____(covariates=____,
values=____)
# Show plot
plt.show()