InferenceModel

Basic class for model inference.

Provides functionality to make predictions and model loading.

Constructors

InferenceModel

Companion

open override fun close()

Closes internal resources: session and kGraph.

graphToString

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 fun predict(inputData: FloatArray): Int

Predicts the class of inputData.

predictSoftly

open fun predictSoftly(inputData: FloatArray, predictionTensorName: String = ""): FloatArray

Predicts vector of probabilities instead of specific class in predict method.

reshape

fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

toString

open override fun toString(): String

isModelInitialized

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.

TrainableModel

SavedModel