r - 有没有办法用 pdp 在 3D 局部绘图上添加 yhat 的颜色

Question

我用这样的包制作了 3D 局部图pdp：

partial1 <- pdp::partial(rf_full, pred.var = c("Long", "Lat"),
                         parallel = T, paropts =list(.packages = "randomForest"))
p2 <- plotPartial(partial1, levelplot = F, colorkey = T, plot.pdp = T)

但是，目标变量 (yhat) 1)的颜色没有填充网格单元格，并且2)似乎总是相同的颜色（对于 yhat 的高值）。它看起来如下：

你知道如何获得这样的东西，例如：

标签： rplotrandom-forest

Answer 1

尝试使用以下选项进行绘图drape=T，该选项是wireframe：

plotPartial(partial1, levelplot = F, colorkey = F, plot.pdp = F, drape=T)