ai.djl.repository.zoo

Class Criteria.Builder<I,O>

java.lang.Object

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

Enclosing class:

Criteria<I,O>

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

A Builder to construct a Criteria.

Method Summary

Modifier and Type	Method and Description
Criteria<I,O>	build() Builds a Criteria instance.
Criteria.Builder<I,O>	optApplication(ai.djl.Application application) Sets the model application for this criteria.
Criteria.Builder<I,O>	optArgument(java.lang.String key, java.lang.Object value) Sets the optional model loading argument for this criteria.
Criteria.Builder<I,O>	optArguments(java.util.Map<java.lang.String,java.lang.Object> arguments) Sets an extra model loading argument for this criteria.
Criteria.Builder<I,O>	optArtifactId(java.lang.String artifactId) Sets optional artifactId of the ModelLoader for this criteria.
Criteria.Builder<I,O>	optDevice(ai.djl.Device device) Sets the Device for this criteria.
Criteria.Builder<I,O>	optEngine(java.lang.String engine) Sets the engine name for this criteria.
Criteria.Builder<I,O>	optFilter(java.lang.String key, java.lang.String value) Sets an extra search filter for this criteria.
Criteria.Builder<I,O>	optFilters(java.util.Map<java.lang.String,java.lang.String> filters) Sets the extra search filters for this criteria.
Criteria.Builder<I,O>	optGroupId(java.lang.String groupId) Sets optional groupId of the ModelZoo for this criteria.
Criteria.Builder<I,O>	optProgress(ai.djl.util.Progress progress) Set the optional Progress.
Criteria.Builder<I,O>	optTranslator(ai.djl.translate.Translator<I,O> translator) Sets the optional Translator to override default Translator.
<P,Q> Criteria.Builder<P,Q>	setTypes(java.lang.Class<P> inputClass, java.lang.Class<Q> outputClass) Creates a new @{code Builder} class with the specified input and output data type.

Methods inherited from class java.lang.Object

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

Method Detail

setTypes

public <P,Q> Criteria.Builder<P,Q> setTypes(java.lang.Class<P> inputClass,
                                            java.lang.Class<Q> outputClass)

Creates a new @{code Builder} class with the specified input and output data type.

Type Parameters:

P - the input data type

Q - the output data type

Parameters:

inputClass - the input class

outputClass - the output class

Returns:

a new @{code Builder} class with the specified input and output data type

optApplication

public Criteria.Builder<I,O> optApplication(ai.djl.Application application)

Sets the model application for this criteria.

Parameters:

application - the model application

Returns:

this Builder

optEngine

public Criteria.Builder<I,O> optEngine(java.lang.String engine)

Sets the engine name for this criteria.

Parameters:

engine - the engine name

Returns:

this Builder

optDevice

public Criteria.Builder<I,O> optDevice(ai.djl.Device device)

Sets the Device for this criteria.

Parameters:

device - the Device for the criteria

Returns:

this Builder

optGroupId

public Criteria.Builder<I,O> optGroupId(java.lang.String groupId)

Sets optional groupId of the ModelZoo for this criteria.

Parameters:

groupId - the groupId of the ModelZoo

Returns:

this Builder

optArtifactId

public Criteria.Builder<I,O> optArtifactId(java.lang.String artifactId)

Sets optional artifactId of the ModelLoader for this criteria.

Parameters:

artifactId - the artifactId of the ModelLoader

Returns:

this Builder

optFilters

public Criteria.Builder<I,O> optFilters(java.util.Map<java.lang.String,java.lang.String> filters)

Sets the extra search filters for this criteria.

Parameters:

filters - the extra search filters

Returns:

this Builder

optFilter

public Criteria.Builder<I,O> optFilter(java.lang.String key,
                                       java.lang.String value)

Sets an extra search filter for this criteria.

Parameters:

key - the search key

value - the search value

Returns:

this Builder

optArguments

public Criteria.Builder<I,O> optArguments(java.util.Map<java.lang.String,java.lang.Object> arguments)

Sets an extra model loading argument for this criteria.

Parameters:

arguments - optional model loading arguments

Returns:

this Builder

optArgument

public Criteria.Builder<I,O> optArgument(java.lang.String key,
                                         java.lang.Object value)

Sets the optional model loading argument for this criteria.

Parameters:

key - the model loading argument key

value - the model loading argument value

Returns:

this Builder

optTranslator

public Criteria.Builder<I,O> optTranslator(ai.djl.translate.Translator<I,O> translator)

Sets the optional Translator to override default Translator.

Parameters:

translator - the override Translator

Returns:

this Builder

optProgress

public Criteria.Builder<I,O> optProgress(ai.djl.util.Progress progress)

Set the optional Progress.

Parameters:

progress - the Progress

Returns:

this Builder

build

public Criteria<I,O> build()

Builds a Criteria instance.

Returns:

the Criteria instance