1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Machine Learning với caret trong R
Connected

Bài tập

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.

Hướng dẫn

100 XP

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.