首页 > 解决方案 > 为什么我的用于交叉验证的 for 循环不断在测试数据上返回相同的预测值?

问题描述

我想查看使用 OLS 对测试数据返回最低 MSE 的交叉验证次数。我创建了一个 for 循环来测试一组交叉验证。具有不同交叉验证次数(5、6、7...)的每个模型的测试集的预测存储在矩阵中。但是,循环继续为每次迭代返回相同的预测集。我预计每组预测都会略有不同。即使手动完成,predict 也会继续为不同数量的交叉验证返回同一组预测值。我错过了什么?任何信息将不胜感激。

set.seed(30122020)
idtrain <- sample(c(1:dim(data_clean)[1]), round(0.8*dim(data_clean)[1]))

train <- data_clean[idtrain,]
test <- data_clean[-idtrain,]
cv_min <- 5
cv_max <- 10
cv_predict <- matrix(, nrow(test), ncol = (cv_max - cv_min + 1))
for (i in cv_min : cv_max) {
  train.control <- trainControl(method = 'cv', number = i)
  OLS.log.cv <- train(logSalePrice ~ ., data = train,
                      method = 'lm', trControl = train.control)
  
  cv_predict[, (i - cv_min + 1)] <- predict(OLS.log.cv, newdata = test)
}
ols_predict

标签: rfor-looptestingregressionpredict

解决方案


推荐阅读