Thuật toán chọn biến tiến dần (forward stepwise) cung cấp một thứ tự tối ưu để thêm biến vào tập biến dự báo. Để quyết định điểm dừng chọn biến, bạn có thể vẽ các đường cong AUC cho train và test. Những đường cong này biểu diễn AUC của train và test khi dùng biến đầu tiên, hai biến đầu tiên, ba biến đầu tiên, … trong mô hình.

Trong bài tập này, bạn sẽ học cách vẽ các đường cong AUC đó. Phương thức auc_train_test để tính AUC đã được cài sẵn và có thể dùng như sau:

auc_train, auc_test = auc_train_test(variables, target, train, test)

trong đó variables là tập biến dùng cho mô hình hồi quy logistic, target là danh sách chứa tên biến mục tiêu, và train cùng test lần lượt là basetable train và test.

Các biến đã được sắp xếp theo thuật toán tiến dần được cung cấp trong danh sách variables. Bạn có thể xem trong console. Ngoài ra, ba danh sách trống đã được định nghĩa sẵn cho bạn:

auc_values_train, sẽ chứa các giá trị AUC trên train của mô hình ở mỗi vòng lặp
auc_values_test, sẽ chứa các giá trị AUC trên test của mô hình ở mỗi vòng lặp
variables_evaluate, sẽ chứa các biến được đánh giá ở mỗi vòng lặp

Chọn biến tiến từng bước (forward stepwise)

Create Your Free Account