Skip to content

Java library and command-line application for converting CatBoost models to PMML

Notifications You must be signed in to change notification settings

jpmml/jpmml-catboost

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

JPMML-CatBoost

Java library and command-line application for converting CatBoost models to PMML.

Prerequisites

  • CatBoost 0.15.2 or newer.
  • Java 1.8 or newer.

Installation

Enter the project root directory and build using Apache Maven:

mvn clean install

The build produces an executable uber-JAR file target/jpmml-catboost-executable-1.0-SNAPSHOT.jar.

Usage

A typical workflow can be summarized as follows:

  1. Use CatBoost to train a model.
  2. Save the model to a CBM file in a local filesystem.
  3. Use the JPMML-CatBoost command-line converter application to turn this CBM file to a PMML file.

The CatBoost side of operations

Using the catboost package to train a regression model for the example Boston housing dataset:

from pandas import DataFrame, Series
from sklearn.datasets import load_boston

boston = load_boston()

X = DataFrame(boston.data, columns = boston.feature_names)
y = Series(boston.target)

from catboost import CatBoostRegressor

cb = CatBoostRegressor(iterations = 131)
cb.fit(X, y)

cb.save_model("catboost.cbm", format = "cbm")

The JPMML-CatBoost side of operations

Converting the CBM file catboost.cbm to a PMML file catboost.pmml:

java -jar target/jpmml-catboost-executable-1.0-SNAPSHOT.jar --cbm-input catboost.cbm --pmml-output catboost.pmml

Getting help:

java -jar target/jpmml-catboost-executable-1.0-SNAPSHOT.jar --help

License

JPMML-CatBoost is dual-licensed under the GNU Affero General Public License (AGPL) version 3.0, and a commercial license.

Additional information

JPMML-CatBoost is developed and maintained by Openscoring Ltd, Estonia.

Interested in using JPMML software in your application? Please contact [email protected]

About

Java library and command-line application for converting CatBoost models to PMML

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published