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

时间:2016-04-14 00:49:30

标签: r xgboost

是否存在已经存在的函数来从R中的xgboost模型获取部分依赖图?我看到了使用mlr包的示例,但它似乎需要一个mlr特定的包装类。我有点不清楚是否有办法将xgboost模型转换为该类。

2 个答案:

答案 0 :(得分:5)

我一直在研究R包pdp,用于构建部分依赖图 - https://github.com/bgreenwell/pdp。我能够使用XGBoost模型,但它仍然是实验性的。手稿文件夹中有一个正在进行的小插图/手稿可能会有所帮助。

答案 1 :(得分:0)