1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Học có giám sát với scikit-learn
Connected

Bài tập

Đường cong ROC

Bây giờ bạn đã xây dựng được mô hình logistic regression để dự đoán tình trạng mắc tiểu đường, bạn có thể vẽ đường cong ROC để trực quan hóa cách tỷ lệ dương tính thật (true positive rate) và tỷ lệ dương tính giả (false positive rate) thay đổi khi ngưỡng quyết định thay đổi.

Nhãn kiểm tra y_test và xác suất dự đoán các đặc trưng kiểm tra thuộc về lớp dương (y_pred_probs) đã được nạp sẵn cho bạn, cùng với matplotlib.pyplot là plt.

Bạn sẽ tạo đường cong ROC và sau đó diễn giải kết quả.

Hướng dẫn 1/2

undefined XP
    1
    2
  • Import roc_curve.
  • Tính các giá trị của đường cong ROC, sử dụng y_test và y_pred_probs, và gán lần lượt vào fpr, tpr, và thresholds.
  • Vẽ tỷ lệ dương tính thật theo tỷ lệ dương tính giả.