io.spokestack.spokestack.tensorflow

Class TensorflowModel.Loader

java.lang.Object

io.spokestack.spokestack.tensorflow.TensorflowModel.Loader

Enclosing class:

TensorflowModel

public static class TensorflowModel.Loader
extends Object

loader (builder) class for the tensorflow model.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TensorflowModel.Loader.DType tensor data types.

Constructor Summary

Constructors

Constructor and Description
Loader() initializes a new loader instance.

Method Summary

Modifier and Type	Method and Description
TensorflowModel	load() loads the tensorflow model using the attached configuration.
TensorflowModel.Loader	reset() resets the loader to the default state.
TensorflowModel.Loader	setPath(String value) sets the file system path to the TF-Lite model.
TensorflowModel.Loader	setStatePosition(int position) sets the position of the model's state tensor in its input array.

Methods inherited from class java.lang.Object

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

Constructor Detail

Loader

public Loader()

initializes a new loader instance.

Method Detail

reset

public TensorflowModel.Loader reset()

resets the loader to the default state.

Returns:

this

setPath

public TensorflowModel.Loader setPath(String value)

sets the file system path to the TF-Lite model.

Parameters:

value - value to assign

Returns:

this

setStatePosition

public TensorflowModel.Loader setStatePosition(int position)

sets the position of the model's state tensor in its input array.

Parameters:

position - the position of the model's state tensor.

Returns:

this

load

public TensorflowModel load()

loads the tensorflow model using the attached configuration.

Returns:

the new tensorflow model