来自R的xgboost模型的部分依赖图

Question

Is there an already existing function to get a partial dependence plot from an xgboost model in R? I saw examples of using mlr package, but it seems to require an mlr -specific wrapper class. I am a little unclear if there is a way to convert an xgboost model into that class.

Answer 1

I have been working on an R package, pdp , for constructing partial dependence plots-- https://github.com/bgreenwell/pdp . I was able to get it to work with XGBoost models, but it is still experimental. There is a vignette/manuscript in progress in the manuscript folder that may help.

Answer 2

这似乎没有转换为mlr ： https ： //github.com/FeiYeYe/xgboost/blob/master/R-package/R/plot.xgb.Booster.R