ai.djl.repository.zoo

Class Criteria<I,O>

java.lang.Object

ai.djl.repository.zoo.Criteria<I,O>

Type Parameters:

I - the model input type

O - the model output type

public class Criteria<I,O>
extends java.lang.Object

The Criteria class contains search criteria to look up a ZooModel.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Criteria.Builder<I,O> A Builder to construct a Criteria.

Method Summary

Modifier and Type	Method and Description
static Criteria.Builder<?,?>	builder() Creates a builder to build a Criteria.
ai.djl.Application	getApplication() Returns the application of the model.
java.util.Map<java.lang.String,java.lang.Object>	getArguments() Returns the override configurations of the model loading arguments.
java.lang.String	getArtifactId() Returns the artifactId of the ModelLoader to be searched.
ai.djl.Device	getDevice() Returns the Device of the model to be loaded on.
java.lang.String	getEngine() Returns the engine name.
java.util.Map<java.lang.String,java.lang.String>	getFilters() Returns the search filters that must match the properties of the model.
java.lang.String	getGroupId() Returns the groupId of the ModelZoo to be searched.
java.lang.Class<I>	getInputClass() Returns the input data type.
java.lang.Class<O>	getOutputClass() Returns the output data type.
ai.djl.util.Progress	getProgress() Returns the optional Progress for the model loading.
ai.djl.translate.Translator<I,O>	getTranslator() Returns the optional Translator to be used for ZooModel.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

getApplication

public ai.djl.Application getApplication()

Returns the application of the model.

Returns:

the application of the model

getInputClass

public java.lang.Class<I> getInputClass()

Returns the input data type.

Returns:

the input data type

getOutputClass

public java.lang.Class<O> getOutputClass()

Returns the output data type.

Returns:

the output data type

getEngine

public java.lang.String getEngine()

Returns the engine name.

Returns:

the engine name

getDevice

public ai.djl.Device getDevice()

Returns the Device of the model to be loaded on.

Returns:

the Device of the model to be loaded on

getGroupId

public java.lang.String getGroupId()

Returns the groupId of the ModelZoo to be searched.

Returns:

the groupId of the ModelZoo to be searched

getArtifactId

public java.lang.String getArtifactId()

Returns the artifactId of the ModelLoader to be searched.

Returns:

the artifactIds of the ModelLoader to be searched

getFilters

public java.util.Map<java.lang.String,java.lang.String> getFilters()

Returns the search filters that must match the properties of the model.

Returns:

the search filters that must match the properties of the model.

getArguments

public java.util.Map<java.lang.String,java.lang.Object> getArguments()

Returns the override configurations of the model loading arguments.

Returns:

the override configurations of the model loading arguments

getTranslator

public ai.djl.translate.Translator<I,O> getTranslator()

Returns the optional Translator to be used for ZooModel.

Returns:

the optional Translator to be used for ZooModel

getProgress

public ai.djl.util.Progress getProgress()

Returns the optional Progress for the model loading.

Returns:

the optional Progress for the model loading

builder

public static Criteria.Builder<?,?> builder()

Creates a builder to build a Criteria.

Returns:

a new builder