是否已经存在从R中的xgboost模型获取部分依赖图的函数?我看到了使用mlr
包的示例,但似乎需要特定于mlr
的包装器类。我还不清楚是否有一种方法可以将xgboost
模型转换为该类。
最佳答案
我一直在研究R包pdp
,用于构造部分依赖图https://github.com/bgreenwell/pdp。我能够使其与XGBoost模型一起使用,但仍处于试验阶段。手稿文件夹中有正在进行中的小插图/手稿可能会有所帮助。
是否已经存在从R中的xgboost模型获取部分依赖图的函数?我看到了使用mlr
包的示例,但似乎需要特定于mlr
的包装器类。我还不清楚是否有一种方法可以将xgboost
模型转换为该类。
最佳答案
我一直在研究R包pdp
,用于构造部分依赖图https://github.com/bgreenwell/pdp。我能够使其与XGBoost模型一起使用,但仍处于试验阶段。手稿文件夹中有正在进行中的小插图/手稿可能会有所帮助。