R caret和gbm无法找到ntrees输入

时间:2012-01-04 05:02:55

标签: r r-caret

我正在尝试使用R中的gbm包来训练caret。我最初得到以下错误并认为这是由于缺少输入,所以我创建了{{1但是我仍然得到相同的错误信息。

gbmGrid

有什么建议吗?

3 个答案:

答案 0 :(得分:5)

predict()函数的正确代码需要从gbmFit $ finalModel对象手动输入.n.trees参数:

    pred1 <- predict(gbmFit$finalModel, newdata = preTest1[,-c(2,60,61,161)], 
              n.trees=gbmFit1$bestTune$.n.trees)

答案 1 :(得分:1)

如果这不起作用:

pred1 <- predict(gbmFit$finalModel, newdata = preTest1[,-c(2,60,61,161)], 
          n.trees=gbmFit1$bestTune$.n.trees)

你可以用这个:

pred1 <- predict(gbmFit, newdata = preTest1[,-c(2,60,61,161)], 
          n.trees=gbmFit1$n.trees)

答案 2 :(得分:0)

我认为您不需要同时传递tuneLengthtuneGrid参数。尝试其中一个,看看问题是否仍然存在。