一、配置JAVA环境和Maven环境
参考以下两个博客即可
MAC 系统安装 JDK 及环境变量配置_=蜗牛=的博客-CSDN博客_mac jdk环境变量配置
MAC 系统安装 Maven 及环境变量配置_=蜗牛=的博客-CSDN博客_mac安装mpv
二、下载JPMML-LightGBM
先在Git上下载
直接下载或使用git clone https://github.com/jpmml/jpmml-lightgbm
下载完后通过命令行 cd 到这个项目下面
然后使用命令行
mvn clean install
如果构建失败,使用以下命令
mvn clean install -DskipTests
或者
mvn clean install -Dmaven.test.skip=true
两个随便选一个就行
然后在这个项目下面的 pmml-lightgbm-example 目录下,会有一个 target 目录
进入target目录,会有几个jar包
使用第3个jar包来进行转换,转换命令如下
java -jar "xx\xx\pmml-lightgbm-example-executable-1.4-SNAPSHOT.jar" --lgbm-input lgb_model.model --pmml-output lgb_model.pmml
jar包的地址改成自己的地址,这里用xx代替
有的博客说模型文件需要是txt格式的,我发现直接使用model格式也可以转换,这个可以根据需要保存成不同的格式来进行转换
三、调用pmml文件
参考[PMML] LightGBM模型保存为PMML文件,通过Java或者Python调用_手撕机的博客-CSDN博客_python lightgbm模型怎么保存为pmml机器学习模型之PMML - 知乎
引用:
- python环境下,XGBoost模型文件转pmml格式用于部署_胖胖大王叫我来巡山的博客-CSDN博客_集群环境下 xgboost模型文件转为pmml文件
- MAC 系统安装 Maven 及环境变量配置_=蜗牛=的博客-CSDN博客_mac安装mpv
- [PMML] LightGBM模型保存为PMML文件,通过Java或者Python调用_手撕机的博客-CSDN博客_python lightgbm模型怎么保存为pmml
- xgb、lgb转pmml - 知乎