Làm thế nào để vẽ đường cong ROC trong Python?

ROC - Đường cong đặc tính hoạt động của máy thu (ROC).

Sử dụng phương pháp metrics.plot_roc_curve (clf, X_test, y_test), chúng ta có thể vẽ đường cong ROC.

Các bước

• Tạo một bài toán phân loại cấp n ngẫu nhiên. Điều này ban đầu tạo ra các cụm điểm được phân phối bình thường (std =1) về các đỉnh của siêu khối chiều `` n_informative '' với các cạnh có độ dài là `` 2 * class_sep '' và chỉ định số lượng cụm bằng nhau cho mỗi lớp.

  Nó giới thiệu sự phụ thuộc lẫn nhau giữa các tính năng này và thêm nhiều loại nhiễu khác vào dữ liệu. Sử dụng phương thức make_classification ().

• Chia mảng hoặc ma trận thành các đoàn tàu ngẫu nhiên, sử dụng phương thức train_test_split ().

• Điều chỉnh mô hình SVM theo dữ liệu đào tạo đã cho, sử dụng phương thức fit ().

• Đường cong đặc tính hoạt động của Bộ thu Plot (ROC), sử dụng phương thức plot_roc_curve ().

• Để hiển thị hình này, hãy sử dụng phương thức plt.show ().

Ví dụ

import matplotlib.pyplot as plt
from sklearn import datasets, metrics, model_selection, svm
X, y = datasets.make_classification(random_state=0)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y, random_state=0)
clf = svm.SVC(random_state=0)
clf.fit(X_train, y_train)
metrics.plot_roc_curve(clf, X_test, y_test)
plt.show()

Đầu ra