xgb和lgb模型文件转pmml环境配置及步骤

xgb和lgb模型文件转pmml环境配置

1、maven环境配置

首先，需要配置好maven环境，用于第二步的jpmml-xgboost和jpmml-lightgbm环境的配置，具体的安装步骤可参考以下教程。

maven的下载安装配置教程

2、jpmml-xgboost和jpmml-lightgbm配置

• 要求java1.8及以上

2.1 jpmml-lightgbm

• 下载

git clone https://github.com/jpmml/jpmml-lightgbm.git

• 配置

下载好project后，通过maven切换到jpmml-lightgbm路径下，然后运行以下代码：

mvn clean install

执行后会生成pmml-lightgbm-example/target/pmml-lightgbm-example-executable-1.5-SNAPSHOT.jar，注意是在pmml-lightgbm-example，后续是用该目录下的程序转pmml。

jpmml-lightgbm下载地址及配置

2.2 jpmml-xgboost

• 下载

git clone https://github.com/jpmml/jpmml-xgboost.git

• 配置

下载好project后，通过maven切换到jpmml-xgboost路径下，然后运行以下代码：

mvn clean install

执行后要确保生成pmml-xgboost-example/target/pmml-xgboost-example-executable-1.7-SNAPSHOT.jar。

jpmml-xgboost下载地址及配置

3、模型转pmml

3.1 lgb转pmml

• 生成模型文件

  lgb转pmml需要.txt文件，即准备好LightGBMAudit.txt文件。

• 模型转pmml

  在jpmml-lightgbm-master\pmml-lightgbm-example路径下打开cmd，然后切换到LightGBMAudit.txt所在路径，执行以下代码：

java -jar pmml-lightgbm-example/target/pmml-lightgbm-example-executable-1.5-SNAPSHOT.jar --lgbm-input LightGBMAudit.txt --pmml-output LightGBMAudit.pmml

3.2 xgb转pmml

• 生成模型文件

  xgb转pmml需要XGBoostAudit.model文件和Audit.fmap文件

• 模型转pmml

  在jpmml-xgboost\pmml-xgboost-example路径下打开cmd，然后切换到XGBoostAudit.model所在路径，执行以下代码：

java -jar pmml-xgboost-example/target/pmml-xgboost-example-executable-1.7-SNAPSHOT.jar --model-input XGBoostAudit.model --fmap-input Audit.fmap --target-name Adjusted --pmml-output XGBoostAudit.pmml