将LightGBM/XGBoost模型加载到sklearn中。

Question

3

LightGBM和XGBoost模型可以转储到包含可读取的模型结构的纯文本文件中。最终，它们只是树集合。

是否有任何库可以将这些转储的模型加载到scikit-learn框架中，例如使用相同的分裂和值构建sklearn集合？

这可能非常方便，因为与sklearn API相关联的一些很好的库，例如treeinterpreter。

- Denis Korzhenkov

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- user2997837 · Accepted Answer

对于XGBoost，您可以使用xgbfir库来解析xgb模型显示的特征交互和排名。安装方法如下：

pip install xgbfir

对于lightGBM，我不知道有什么好的选项。微软的lightGBM库允许PMML导出，因此您可以导出然后使用一些PMML解析器。