交叉驗證

使用 R 的 caret 進行 Machine Learning

Zach Mayer

Data Scientist at DataRobot and co-author of caret

交叉驗證

在完整資料集上擬合最終模型

交叉驗證

# Set seed for reproducibility
set.seed(42)

# Fit linear regression model
model <- train(
  mpg ~ hp, mtcars, 
  method = "lm",
  trControl = trainControl(
    method = "cv", 
    number = 10,
    verboseIter = TRUE
  )
)

+ Fold01: intercept=TRUE 
- Fold01: intercept=TRUE 
...
- Fold10: intercept=TRUE 
Aggregating results
Fitting final model on full training set

一起來練習吧！

使用 R 的 caret 進行 Machine Learning