r - R の xgboost モデルからの部分依存プロット

Question

Rのxgboostモデルから部分依存プロットを取得する既存の関数はありますか? パッケージの使用例を見ましたが、固有のラッパー クラスmlrが必要なようです。モデルをそのクラスmlrに変換する方法があるかどうかは少しわかりません。xgboost

Answer 1

pdp私は、部分依存プロットを構築するためのRパッケージに取り組んできました-- https://github.com/bgreenwell/pdp。XGBoost モデルで動作させることができましたが、まだ実験段階です。原稿フォルダーに進行中のビネット/原稿があり、役立つ可能性があります。

Answer 2

これは、変換せずにトリックを行うようですmlr: https://github.com/FeiYeYe/xgboost/blob/master/R-package/R/plot.xgb.Booster.R