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

最佳答案

我一直在研究R包pdp,用于构造部分依赖图https://github.com/bgreenwell/pdp。我能够使其与XGBoost模型一起使用,但仍处于试验阶段。手稿文件夹中有正在进行中的小插图/手稿可能会有所帮助。

10-06 11:32