是否可以绘制部分依赖图以显示类概率并估计GBM模型的预测变量的影响？类似于partialPlotfrom randomForestpackage 的东西。
根据这篇文章，使用 gbm 可以进行部分绘图。
在此先感谢您的帮助。
部分依赖图和梯度提升（GBM 包）
机器算法验证
r
助推
2022-03-09 11:04:16
1个回答
是的，这真的很容易。退房?plot.gbm。用法是这样的：
plot(gbm.model, i.var = 1, lwd = 2, col = "blue", main = "")
看起来像
i.var控制要绘制的变量的索引。您还可以使用两种（或更多）方式绘图
plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")