我正在尝试使用插入符号来找到 gbm 模型的最佳参数。此代码与我在其他数据集上使用的代码相同,无法找出错误。
它似乎可以运行模型,但无法创建预测。
predictions failed for Fold2: interaction.depth=4, shrinkage=0.005, n.trees=200 Error in apply(tmp, 2, function(x, nm = modelFit$obsLevels) ifelse(x >= :
dim(X) must have a positive length
这是完整的代码:
library(caret)
library(gbm)
myControl <- trainControl(method='cv', number=2, summaryFunction=twoClassSummary,
classProbs=TRUE, savePredictions=TRUE, verboseIter=TRUE)
df1 <- data.frame(Y = round(runif(1000), 0), x1=runif(1000), x2=runif(1000) )
X <- df1[,c('x1','x2')]
Y <- factor(paste('X', df1[,'Y']))
gbm_model <- train(X, Y, method='gbm', metric='ROC', trControl=myControl
,distribution='bernoulli', tuneGrid=expand.grid(.n.trees=seq(100, 200, by=100)
,.interaction.depth=seq(2, 4, by=2), .shrinkage=c(.005)))
有什么建议么?
编辑:我正在使用gbm 2.1
和caret 5.16.24