Java library and command-line application for converting LightGBM models to PMML.
- Java 1.8 or newer.
Enter the project root directory and build using Apache Maven:
mvn clean install
The build produces an executable uber-JAR file target/converter-executable-1.2-SNAPSHOT.jar
.
A typical workflow can be summarized as follows:
- Use LightGBM to train a model.
- Save the model to a text file in a local filesystem.
- Use the JPMML-LightGBM command-line converter application to turn this text file to a PMML file.
Using the lightgbm
package to train a regression model for the example boston housing dataset:
from sklearn.datasets import load_boston
boston = load_boston()
from lightgbm import LGBMRegressor
lgbm = LGBMRegressor(objective = "regression")
lgbm.fit(boston.data, boston.target, feature_name = boston.feature_names)
lgbm.booster_.save_model("lightgbm.txt")
Converting the text file lightgbm.txt
to a PMML file lightgbm.pmml
:
java -jar target/converter-executable-1.2-SNAPSHOT.jar --lgbm-input lightgbm.txt --pmml-output lightgbm.pmml
Getting help:
java -jar target/converter-executable-1.2-SNAPSHOT.jar --help
JPMML-LightGBM is licensed under the GNU Affero General Public License (AGPL) version 3.0. Other licenses are available on request.
Please contact [email protected]