TensorFlowInferenceModel

open class TensorFlowInferenceModel : InferenceModel

Basic class for model inference.

Provides functionality to make predictions and model loading.

Constructors

fun TensorFlowInferenceModel()

Types

Companion

object Companion

Functions

open override fun close()

Closes internal resources: session and kGraph.

copy

open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): TensorFlowInferenceModel

graphToString

fun graphToString(): String

Forms the graph description in string format.

input

fun input(inputOp: Input)

Chain-like setter to set up inputOp.

output

fun output(outputOp: Output)

Chain-like setter to set up outputOp.

predict

open override fun predict(inputData: FloatArray): Int

Generates output prediction for the input sample.

predictSoftly

open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray

reshape

open override fun reshape(vararg dims: Long)

toString

open override fun toString(): String

Properties

inputDimensions

open override val inputDimensions: LongArray

isModelInitialized

var isModelInitialized: Boolean = false

Is true when model is initialized.

kGraph

lateinit var kGraph: KGraph

TensorFlow wrapped computational graph.

name

var name: String? = null

Model name.

shape

lateinit var shape: LongArray

Data shape for prediction.

Inheritors

TrainableModel

SavedModel