Skip to content
This repository has been archived by the owner on Apr 10, 2019. It is now read-only.
/ jpmml-tensorflow Public archive

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

License

Notifications You must be signed in to change notification settings

jpmml/jpmml-tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

JPMML-TensorFlow

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

Features

Prerequisites

The TensorFlow side of operations

  • Protocol Buffers 3.2.0 or newer
  • TensorFlow 1.1.0 or newer

The Java side of operations

  • Java 1.8 or newer

Installation

Enter the project root directory and build using Apache Maven; use the protoc.exe system property to specify the location of the Protocol Buffers compiler:

mvn -Dprotoc.exe=/usr/local/bin/protoc clean install

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

Usage

A typical workflow can be summarized as follows:

  1. Use TensorFlow to train an estimator.
  2. Export the estimator in SavedModel data format to a directory in a local filesystem.
  3. Use the JPMML-TensorFlow command-line converter application to turn the SavedModel directory to a PMML file.

The TensorFlow side of operations

Please see the test script file main.py for sample workflows.

The Java side of operations

Converting the estimator SavedModel directory estimator/ to a PMML file estimator.pmml:

java -jar target/converter-executable-1.0-SNAPSHOT.jar --tf-savedmodel-input estimator/ --pmml-output estimator.pmml

Getting help:

java -jar target/converter-executable-1.0-SNAPSHOT.jar --help

License

JPMML-TensorFlow is licensed under the GNU Affero General Public License (AGPL) version 3.0. Other licenses are available on request.

Additional information

Please contact [email protected]

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published