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

Question

我用 pdp 包制作了 3D 局部图，如下所示：

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）。它看起来如下：

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

Answer 1

尝试使用以下选项 drape=T 进行绘图，这是 wireframe 的一个选项：

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

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

Is there a way to add color of yhat on 3D partial plot with pdp

r

plot

random-forest