Package ai.djl.huggingface.translator

Class ZeroShotClassificationTranslator.Builder

java.lang.Object

ai.djl.huggingface.translator.ZeroShotClassificationTranslator.Builder

Enclosing class:

ZeroShotClassificationTranslator

public static final class ZeroShotClassificationTranslator.Builder
extends Object

The builder for zero-shot classification translator.

Method Summary

Modifier and Type	Method	Description
ZeroShotClassificationTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
ZeroShotClassificationTranslator.Builder	optInt32(boolean int32)	Sets if use int32 datatype for the Translator.

Methods inherited from class java.lang.Object

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

Method Details

optInt32

public ZeroShotClassificationTranslator.Builder optInt32(boolean int32)

Sets if use int32 datatype for the Translator.

Parameters:

int32 - true to include token types

Returns:

this builder

configure

public void configure(Map<String,?> arguments)

Configures the builder with the model arguments.

Parameters:

arguments - the model arguments

build

public ZeroShotClassificationTranslator build()
                                       throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs