0

我在 R 中使用 caret 的 train 函数来生成使用 GBM 的模型。我使用了 5 次重复的重复交叉验证,这意味着将有 50 个样本。我想问是否有一种方法可以以不同的方式绘制结果,以便该图在 x 轴上显示提升迭代,在 y 轴上显示 auc,并且在其内部显示从最佳参数选择获得的结果,但是用于训练折叠和测试折叠的单独行。这可以在您使用 gbm 包中的“gbm”函数并使用“gbm.perf”以及采样技术来绘制偏差的训练和验证曲线时产生。

是否有可能以某种方式对插入符号的火车功能做同样的事情?

谢谢。

4

1 回答 1

1

在插入符号对象中,如果使用 method='gbm',则可以选择属性“finalModel”,即生成的 gbm 对象。例如,如果您的火车对象被命名为“a”,那么

gbm_model <- a$finalModel

使用 gbm_model,您可以运行 gbm 包内部的函数。

于 2015-10-02T19:17:10.083 回答