标签: r xgboost
是否存在已经存在的函数来从R中的xgboost模型获取部分依赖图?我看到了使用mlr包的示例,但它似乎需要一个mlr特定的包装类。我有点不清楚是否有办法将xgboost模型转换为该类。
mlr
xgboost
答案 0 :(得分:5)
我一直在研究R包pdp,用于构建部分依赖图 - https://github.com/bgreenwell/pdp。我能够使用XGBoost模型,但它仍然是实验性的。手稿文件夹中有一个正在进行的小插图/手稿可能会有所帮助。
pdp
答案 1 :(得分:0)
这似乎没有转换为mlr:https://github.com/FeiYeYe/xgboost/blob/master/R-package/R/plot.xgb.Booster.R