@ProviderType
public class ModelInfo
extends java.lang.Object
Modifier | Constructor and Description |
---|---|
protected |
ModelInfo(java.lang.String modelName,
java.util.Optional<java.lang.String> platform,
java.util.Optional<java.lang.String> version,
java.util.Map<java.lang.String,java.lang.Object> parameters,
java.util.List<TensorDescriptor> inputDescriptors,
java.util.List<TensorDescriptor> outputDescriptors)
Instantiates a new ModelInfo
|
Modifier and Type | Method and Description |
---|---|
static ModelInfoBuilder |
builder(java.lang.String name)
Instantiates a builder for a
ModelInfo |
java.util.List<TensorDescriptor> |
getInputs()
Return the descriptors of the input tensors
|
java.lang.String |
getName()
Return the name of the model
|
java.util.List<TensorDescriptor> |
getOutputs()
Return the descriptors of the output tensors
|
java.util.Map<java.lang.String,java.lang.Object> |
getParameters()
Return the optional parameters assigned to the model
|
java.util.Optional<java.lang.String> |
getPlatform()
Return the platform used for running this model
|
java.util.Optional<java.lang.String> |
getVersion()
Return the version of the model
|
ModelInfoBuilder |
toBuilder()
Creates a new
ModelInfoBuilder and initializes it from this ModelInfo instance. |
protected ModelInfo(java.lang.String modelName, java.util.Optional<java.lang.String> platform, java.util.Optional<java.lang.String> version, java.util.Map<java.lang.String,java.lang.Object> parameters, java.util.List<TensorDescriptor> inputDescriptors, java.util.List<TensorDescriptor> outputDescriptors)
modelName
- the name of the modelplatform
- an optional string representing the model platformversion
- an optional string representing the model versionparameters
- a map containing the model parameters. It can be empty.inputDescriptors
- a list of TensorDescriptor
of the input tensorsoutputDescriptors
- a list of TensorDescriptor
of the output tensorspublic static ModelInfoBuilder builder(java.lang.String name)
ModelInfo
name
- the name of the modelModelInfoBuilder
public ModelInfoBuilder toBuilder()
ModelInfoBuilder
and initializes it from this ModelInfo
instance.ModelInfoBuilder
public java.lang.String getName()
public java.util.Optional<java.lang.String> getPlatform()
public java.util.Optional<java.lang.String> getVersion()
public java.util.Map<java.lang.String,java.lang.Object> getParameters()
public java.util.List<TensorDescriptor> getInputs()
TensorDescriptor
of the input tensorspublic java.util.List<TensorDescriptor> getOutputs()
TensorDescriptor
of the output tensors