ModelType

interface ModelType<T : InferenceModel, U : InferenceModel>
Content copied to clipboard

Basic interface for models loaded from S3.

Parameters

T

the type of the basic model for common functionality.

U

the type of the pre-trained model for usage in Easy API.

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): T
Content copied to clipboard

Loads the model, identified by this name, from the modelHub.

pretrainedModel
Link copied to clipboard
abstract fun pretrainedModel(modelHub: ModelHub): U
Content copied to clipboard

Returns the specially prepared pre-trained model of the type U.

Properties

modelRelativePath
Link copied to clipboard
abstract val modelRelativePath: String
Content copied to clipboard

Relative path to model for local and S3 buckets storages.

preprocessor
Link copied to clipboard
open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
Content copied to clipboard

Preprocessing Operation specific for this model type.