一、lgb模型保存为pmml文件
1.首先需要下载jdk并将环境配置好,可参考:
https://www.cnblogs.com/pengpengdeyuan/p/14212752.html
2.再将maven下载配置好,可参考:
https://www.cnblogs.com/pengpengdeyuan/p/14217772.html
3.下载此文件并解压,下载地址:
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