部分依赖/ALE/ICE 在 r 中绘制 XgBoost答案

【问题标题】：Partial dependence/ALE/ICE plots XgBoost in r部分依赖/ALE/ICE 在 r 中绘制 XgBoost
【发布时间】：2021-05-07 00:52:52
【问题描述】：

我正在尝试为使用 Xgboost 库构建的 r 中的回归 Xgboost 模型绘制 pdp、ale 和 ICE 图。我已经尝试过使用 pdp 库：

library(pdp)

    xv <- data.matrix(subset(data, select = -ICP))  # training features
    p1xv <- partial(xgbc, pred.var = "za1", ice = TRUE, center = TRUE, 
    plot = TRUE, rug = TRUE, alpha = 0.1, plot.engine = "ggplot2", train = xv)

我收到以下错误：

partial.default(xgbc, pred.var = "za1", ice = TRUE, center = TRUE, 中的错误：部分依赖值目前仅适用于分类和回归问题。

虽然该模型是功能性的，但我设法使用 modelstudio 绘制了故障图。关于错误原因的任何想法？模型中是否有需要专门定义的参数来生成这些图。 za1 是一个数值变量。

【问题讨论】：

标签： xgboost pdp

【解决方案1】：

您需要指定类型。如果ICP是连续的，试试

p1xv <- partial(xgbc, pred.var = "za1", ice = TRUE, center = TRUE, plot = TRUE, rug = TRUE, alpha = 0.1, plot.engine = "ggplot2", train = xv, type = "regression")

【讨论】：