Machine Learning với caret trong R
Zach Mayer
Data Scientist at DataRobot and co-author of caret
?preProcess để biết thêm# Tạo dữ liệu có giá trị khuyết
data(mtcars)
set.seed(42)
mtcars[sample(1:nrow(mtcars), 10), "hp"] <- NA
Y <- mtcars$mpg
X <- mtcars[,2:4] # <- Thiếu ngẫu nhiên
# Dùng "công thức" cho mô hình tuyến tính
set.seed(42)
model <- train(
X, Y, method = "glm",
preProcess = c("center", "scale", "medianImpute")
)
print(min(model$results$RMSE))
3.612713
# PCA trước khi mô hình hóa
set.seed(42)
model <- train(
X, Y, method = "glm",
preProcess = c("center", "scale", "medianImpute", "pca")
)
min(model$results$RMSE)
3.402557
# Biến đổi spatial sign
set.seed(42)
model <- train(
X, Y, method = "glm",
preProcess = c("center", "scale", "medianImpute", "spatialSign")
)
min(model$results$RMSE)
4.284904
Machine Learning với caret trong R