Package | Description |
---|---|
org.apache.ignite.ml.inference.parser |
Root package for model inference parsers.
|
Modifier and Type | Class and Description |
---|---|
class |
TensorFlowGraphModelParser<I,O>
Implementation of TensorFlow model parser that accepts serialized graph definition.
|
class |
TensorFlowSavedModelModelParser<I,O>
Implementation of TensorFlow model parser that accepts serialized directory with "SavedModel" as an input.
|
Modifier and Type | Method and Description |
---|---|
TensorFlowBaseModelParser<I,O> |
TensorFlowBaseModelParser.withInput(String name,
TensorFlowBaseModelParser.InputTransformer<I> transformer)
Setter that allows to specify additional input graph node and correspondent transformer that allows to transform
input into tensor.
|
TensorFlowBaseModelParser<I,O> |
TensorFlowBaseModelParser.withOutput(List<String> names,
TensorFlowBaseModelParser.OutputTransformer<O> transformer)
Setter that allows to specify output graph nodes and correspondent transformer that allow to transform tensors
into output.
|
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024