Tạo một trainControl tùy chỉnh

Bộ dữ liệu chất lượng rượu vang là một bài toán hồi quy, nhưng bây giờ bạn sẽ xem một bài toán phân loại. Đây là một bộ dữ liệu mô phỏng dựa trên cuộc thi "don't overfit" trên Kaggle cách đây vài năm.

Bài toán phân loại phức tạp hơn hồi quy một chút vì bạn cần cung cấp summaryFunction tùy chỉnh cho hàm train() để dùng chỉ số AUC xếp hạng các mô hình. Hãy bắt đầu bằng cách tạo một trainControl tùy chỉnh, giống như bạn đã làm ở chương trước. Nhớ đặt classProbs = TRUE, nếu không twoClassSummary trong summaryFunction sẽ không chạy được.

Bài tập này là một phần của khóa học

Machine Learning với caret trong R

Xem khóa học

Hướng dẫn bài tập

Tạo một trainControl tùy chỉnh tên myControl cho phân loại bằng hàm trainControl.

Dùng 10 nếp gấp CV.
Dùng twoClassSummary cho summaryFunction.
Nhớ đặt classProbs = TRUE.

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.

# Create custom trainControl: myControl
myControl <- trainControl(
  method = "cv", 
  number = ___,
  summaryFunction = ___,
  classProbs = ___, # IMPORTANT!
  verboseIter = TRUE
)

Chỉnh sửa và Chạy Mã