[英]Partial dependence/ALE/ICE plots XgBoost in r
我正在嘗試使用 Xgboost 庫構建的 r 中的 plot pdp、ale 和 ICE 繪制回歸 Xgboost model。 我已經嘗試過使用 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, : 部分依賴值目前僅可用於分類和回歸問題。
盡管 model 可以正常工作,但我使用 modelstudio 設法對 plot 進行了分解圖。 關於錯誤原因的任何想法? model 中是否有需要專門定義以生成這些圖的參數。 za1 是一個數值變量。
您需要指定類型。 如果 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")
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.