Đường cong ROC
Ngoài ngưỡng 0.5, bạn cũng có thể chọn các giá trị cut-off khác, nhưng số lượng cut-off tiềm năng rất lớn và phải xem hàng trăm bảng sẽ tốn nhiều thời gian. Tạo đường cong ROC sẽ giúp bạn.
Bạn có thể tạo biểu đồ này bằng hàm roc() từ gói bổ sung pROC. Hàm roc() nhận đầu vào là vector các phản hồi quan sát (thường mã hóa dưới dạng 0 và 1) và một vector các giá trị dự đoán có cùng độ dài. Một lần nữa, bạn lấy các lượt mua HOPPINESS đã quan sát từ đối tượng choice.data. Các giá trị dự đoán thu được bằng cách áp dụng hàm fitted() lên đối tượng extended.model. Đường cong ROC tương ứng được tạo bằng cách áp dụng hàm plot() lên đối tượng roc thu được.
Bài tập này là một phần của khóa học
Xây dựng mô hình phản hồi trong R
Hướng dẫn bài tập
- Nạp gói bổ sung
pROCbằng hàmlibrary(). - Lấy các lượt mua đã quan sát cho
HOPPINESStừchoice.datavà gán vào một đối tượngobserved. - Tạo một đối tượng
rocbằng cách dùng hàmroctrên các phản hồi quan sát trongchoice.datavà các giá trị dự đoán lấy từextended.model. Gán kết quả vào một đối tượng tênROC. - Vẽ đường cong ROC bằng cách dùng hàm
plot()trên đối tượngROC.
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.
# Load the pROC package
# Obtain the observed purchases
observed <- ___
# Create the Roc object
ROC <- roc(predictor = ___, response = ___)
# Plot the ROC curve