set.seed(1234)
split <- createDataPartition(
y = chem$Yield,
p = 0.8,
list = FALSE
)
train_chem <- chem[split,]
test_chem <- chem[-split,]
ctrl <- trainControl(method = 'repeatedcv', number = 10, repeats = 5)
lambdas <- 10^seq(3, -3, length.out = 100)
set.seed(1234)
chem_ridge_model <- train(
x = train_chem |> select(-Yield),
y = train_chem$Yield,
method = "glmnet",
trControl = ctrl,
preProcess = c('zv', 'knnImpute'),
tuneGrid = expand.grid(alpha = 0, lambda = lambdas),
lambda = lambdas
)
chem_ridge_model$bestTune