是否可以绘制部分依赖关系图以显示类概率并估计预测变量对GBM模型的影响?
类似于partialPlot
包中randomForest
的内容。
根据this article,部分情节可以与gbm一起使用。
提前感谢您的帮助。
答案 0 :(得分:4)
我找到了解决方案,
为了估计预测变量的影响,我们应该使用它:
plot(gbm.model, i.var = 1, lwd = 2, main = "")
i.var
是变量预测变量的编号,按公式中显示的顺序排列。
E.g y = x1+x2+x3
感谢坎贝尔先生