InferenceModel

interface InferenceModel : AutoCloseable

The basic interface for all models which defines the basic functions required for inference tasks only.

Functions

abstract fun close()

copy

abstract fun copy(copiedModelName: String? = null, saveOptimizerState: Boolean = false, copyWeights: Boolean = true): InferenceModel

Creates a copy.

predict

abstract fun predict(inputData: FloatArray): Int

Predicts the class of inputData.

predictSoftly

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

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

reshape

abstract fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

inputDimensions

Input specification for this model.