LightGBM 和 XGBoost 模型可以转储为包含人类可读模型结构的纯文本文件。最后，它们只是树的集合。
是否有任何库可以将这些转储模型加载到 scikit-learn 框架中，例如构建具有相同拆分和值的 sklearn 集成？
这可能很方便，因为 sklearn API 附带了一些不错的库，例如treeinterpreter。
问问题
877 次
1 回答
1
对于 XGBoost，您可以使用xgbfir 库来解析 xgb 模型显示特征交互和排名。安装它：
pip install xgbfir
对于 lightGBM，我不知道有什么好的选择。Microsoft 的 lightGBM 库允许 PMML 导出，因此也许您可以导出然后使用一些 PMML 解析器。
于 2018-01-10T21:54:27.513 回答