lightgbm模型保存为pmml文件,并加载

一、lgb模型保存为pmml文件

1.首先需要下载jdk并将环境配置好,可参考:

https://www.cnblogs.com/pengpengdeyuan/p/14212752.html

2.再将maven下载配置好,可参考:

https://www.cnblogs.com/pengpengdeyuan/p/14217772.html

3.下载此文件并解压,下载地址:

GitHub - jpmml/jpmml-lightgbm: Java library and command-line application for converting LightGBM models to PMML

4.在解压后的文件夹目录下输入:

mvn clean install

5.在pmml-lightgbm-example目录下会生成target文件,target目录下会生成.jar文件

6.进入target目录下,将你的.txt文件放入此目录中,输入以下命令:

java -jar "\...\target\pmml\pmml-lightgbm-example-executable-1.5-SNAPSHOT.jar" --lgbm-input model.txt --pmml-output model.pmml
# 注意\...\中更改为你的路径
# .txt文件更改为你的文件名称
# .jar文件注意你生成的版本号

7.在此目录下可生成.pmml文件

二、python环境下pmml模型的加载

1.下载sklearn2pmml和pypmml

pip install sklearn2pmml -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pypmml -i https://pypi.tuna.tsinghua.edu.cn/simple

2.导入以及模型加载和使用

from sklearn2pmml import PMMLPipeline,sklearn2pmml
from pypmml import Model
m = Model.fromFile('model.pmml')
m.predict(data)

参考:

xgb和lgb模型文件转pmml环境配置及步骤_lightgbm pmml-CSDN博客

https://zhuanlan.zhihu.com/p/265888201

如何将lightgbm模型存为pmml文件_lightgbm转pmml-CSDN博客

Python如何保存和加载PMML文件_python加载pmml-CSDN博客

  • 5
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值