使用插入符号列车功能生成训练和验证集的图形

Question

我在R中使用插入符号列车功能来生成使用GBM的模型。我使用了5次重复的重复交叉验证，这意味着将有50个样本。我想问一下是否有办法以不同的方式绘制结果，使得图表显示x轴上的增强迭代和y轴上的auc，内部显示从最佳参数选择获得的结果但是用于训练折叠和测试折叠的单独线。当您使用gbm包中的“gbm”函数并使用“gbm.perf”以及采样技术绘制偏差的训练和验证曲线时，可以生成此信息。

是否有可能以某种方式对插入符号的列车功能进行相同的操作？

感谢。

Answer 1

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

gbm_model＆lt; - $ finalModel

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