部分依赖图和梯度提升（GBM 包）

Posted 2023-03-12

【中文标题】部分依赖图和梯度提升（GBM 包）

【英文标题】：Partial Dependency plots and Gradient boosting (GBM package)

【发布时间】：2013-10-15 10:42:06

【问题描述】：

是否可以绘制部分依赖图以显示类概率并估计 GBM 模型的预测变量的影响？ 类似于 partialPlot 的东西来自 randomForest 包。

根据this article，gbm 可以进行部分绘图。

提前感谢您的帮助。

【参考方案1】：

我找到了解决办法，

为了估计预测器的效果，应该使用这个：

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

i.var 是变量预测器的编号，按照公式中显示的顺序。 例如y = x1+x2+x3

感谢坎贝尔先生

【讨论】：

如果你也想使用实际的变量名，也可以使用i="variable name"