gpt4 book ai didi

scikit-learn - 将 LightGBM/XGBoost 模型加载到 sklearn

转载 作者:行者123 更新时间:2023-12-04 17:48:15 27 4
gpt4 key购买 nike

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

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

这可能非常方便,因为有一些不错的库附加到 sklearn API,例如treeinterpreter .

最佳答案

对于 XGBoost,您可以使用 xgbfir library它解析 xgb 模型显示特征交互和排名。安装它:

pip install xgbfir

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

关于scikit-learn - 将 LightGBM/XGBoost 模型加载到 sklearn,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47285315/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com