org.apache.ignite.ml.inference.parser

Class TensorFlowSavedModelModelParser<I,O>

java.lang.Object

org.apache.ignite.ml.inference.parser.TensorFlowBaseModelParser<I,O>

org.apache.ignite.ml.inference.parser.TensorFlowSavedModelModelParser<I,O>

Type Parameters:

I - Type of model input.

O - Type of model output.

All Implemented Interfaces:

Serializable, ModelParser<I,O,Model<I,O>>

public class TensorFlowSavedModelModelParser<I,O>
extends TensorFlowBaseModelParser<I,O>

Implementation of TensorFlow model parser that accepts serialized directory with "SavedModel" as an input. The directory is assumed to be serialized by DirectorySerializer.

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.ignite.ml.inference.parser.TensorFlowBaseModelParser

TensorFlowBaseModelParser.InputTransformer<I>, TensorFlowBaseModelParser.OutputTransformer<O>

Constructor Summary

Constructors

Constructor and Description
TensorFlowSavedModelModelParser(String... tags) Constructs a new instance of TensorFlow model parser.

Method Summary

Modifier and Type	Method and Description
org.tensorflow.Session	parseModel(byte[] mdl) Parses model specified in serialized form as byte array.

Methods inherited from class org.apache.ignite.ml.inference.parser.TensorFlowBaseModelParser

parse, withInput, withOutput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TensorFlowSavedModelModelParser

public TensorFlowSavedModelModelParser(String... tags)

Constructs a new instance of TensorFlow model parser.

Parameters:

tags - Model tags.

Method Detail

parseModel

public org.tensorflow.Session parseModel(byte[] mdl)

Parses model specified in serialized form as byte array.

Specified by:

parseModel in class TensorFlowBaseModelParser<I,O>

Parameters:

mdl - Inference model in serialized form as byte array.

Returns:

TensorFlow session that encapsulates the TensorFlow graph parsed from serialized model.