使用自訂的 trainControl
現在你已經有自訂的 trainControl 物件,就能輕鬆訓練以 AUC 而非準確率來調參與評估的 caret 模型。你只要透過 trControl 參數把自訂的 trainControl 物件傳給 train(),例如:
train(<standard arguments here>, trControl = myControl)
這種語法讓你可以方便地保存大量自訂的建模參數,接著在多次呼叫 train() 時重複使用。你會在第 5 章大量運用這個技巧。
本練習屬於課程
使用 R 的 caret 進行 Machine Learning
練習說明
- 使用
train()來以Sonar資料中的其他所有變數預測Class(也就是Class ~ .)。模型應為glm(也就是將method設為"glm"),並使用你自訂的trainControl物件myControl。將結果存到model。 - 在主控台列印該模型並檢視輸出。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Train glm with custom trainControl: model
# Print model to console