6

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

4

2 回答 2

6

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

于 2016-09-07T19:39:44.690 回答
-1

这似乎可以解决问题而无需转换为mlrhttps ://github.com/FeiYeYe/xgboost/blob/master/R-package/R/plot.xgb.Booster.R

于 2016-04-14T02:50:51.703 回答