@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)
ModelInfoname - the name of the modelModelInfoBuilderpublic ModelInfoBuilder toBuilder()
ModelInfoBuilder and initializes it from this ModelInfo instance.ModelInfoBuilderpublic 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