Package ai.djl.huggingface.translator

Class TextClassificationTranslator.Builder

java.lang.Object

ai.djl.huggingface.translator.TextClassificationTranslator.Builder

Enclosing class:

TextClassificationTranslator

public static final class TextClassificationTranslator.Builder
extends Object

The builder for token classification translator.

Method Summary

Modifier and Type	Method	Description
TextClassificationTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
TextClassificationTranslator.Builder	optBatchifier(ai.djl.translate.Batchifier batchifier)	Sets the Batchifier for the Translator.
TextClassificationTranslator.Builder	optIncludeTokenTypes(boolean includeTokenTypes)	Sets if include token types for the Translator.

Methods inherited from class java.lang.Object

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

Method Details

optIncludeTokenTypes

public TextClassificationTranslator.Builder optIncludeTokenTypes(boolean includeTokenTypes)

Sets if include token types for the Translator.

Parameters:

includeTokenTypes - true to include token types

Returns:

this builder

optBatchifier

public TextClassificationTranslator.Builder optBatchifier(ai.djl.translate.Batchifier batchifier)

Sets the Batchifier for the Translator.

Parameters:

batchifier - 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 TextClassificationTranslator build()
                                   throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs