部分依赖关系图和梯度提升(GBM包)

时间:2013-10-08 13:22:22

标签: r gbm

是否可以绘制部分依赖关系图以显示类概率并估计预测变量对GBM模型的影响? 类似于partialPlot包中randomForest的内容。

enter image description here

根据this article,部分情节可以与gbm一起使用。

提前感谢您的帮助。

1 个答案:

答案 0 :(得分:4)

我找到了解决方案,

为了估计预测变量的影响,我们应该使用它:

plot(gbm.model, i.var = 1, lwd = 2, main = "")

i.var是变量预测变量的编号,按公式中显示的顺序排列。 E.g y = x1+x2+x3

感谢坎贝尔先生