InferenceModel

open class InferenceModel : AutoCloseable

Basic class for model inference.

Provides functionality to make predictions and model loading.

Constructors

InferenceModel

fun InferenceModel()

Types

Companion

object Companion

Functions

open override fun close()

Closes internal resources: session and kGraph.

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 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

Properties

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

InferenceModel

Constructors

On this page

On this page

Types

On this page

On this page

Functions

On this page

On this page

Properties

On this page

On this page

Inheritors

On this page

On this page