将LightGBM / XGBoost模型加载到sklearn

时间:2017-11-14 11:59:35

标签: scikit-learn xgboost lightgbm

LightGBM和XGBoost模型可以转储到包含人类可读模型结构的纯文本文件中。最后,它们只是树组合。

是否有任何库可以将这些转储模型加载到scikit-learn框架,例如构造具有相同分裂和值的sklearn集合?

这可能很安静,因为有一些很好的库附加到sklearn API,例如treeinterpreter

1 个答案:

答案 0 :(得分:1)

对于XGBoost,您可以使用xgbfir library解析xgb模型显示功能的交互和排名。安装时使用:

pip install xgbfir

对于lightGBM,我不知道好的选择。 Microsoft的lightGBM库允许PMML导出,因此您可以导出然后使用一些PMML解析器。