Partial graph of xgboost model in R

Question

Is there an existing function to get a partial dependency plot on the xgboost model in R? I have seen examples of using the mlr package, but it requires a class of mlr specific shells. I am a bit unclear if there is a way to convert the xgboost model to this class.

Answer 1

I worked on the R, pdp package to build partial dependency sites - https://github.com/bgreenwell/pdp . I managed to get it to work with XGBoost models, but it is still experimental. There is a vignette / manuscript file in the manuscript folder.

Answer 2

This seems like a trick without conversion to mlr : https://github.com/FeiYeYe/xgboost/blob/master/R-package/R/plot.xgb.Booster.R