LightGBM和XGBoost模型可以转储到包含人类可读模型结构的纯文本文件中。最后,它们只是树组合。
是否有任何库可以将这些转储模型加载到scikit-learn框架,例如构造具有相同分裂和值的sklearn集合?
这可能很安静,因为有一些很好的库附加到sklearn API,例如treeinterpreter
答案 0 :(得分:1)
对于XGBoost,您可以使用xgbfir library解析xgb模型显示功能的交互和排名。安装时使用:
pip install xgbfir
对于lightGBM,我不知道好的选择。 Microsoft的lightGBM库允许PMML导出,因此您可以导出然后使用一些PMML解析器。