Package ai.djl.huggingface.translator

Class CrossEncoderTranslator.Builder

java.lang.Object

ai.djl.huggingface.translator.CrossEncoderTranslator.Builder

Enclosing class:

CrossEncoderTranslator

public static final class CrossEncoderTranslator.Builder
extends Object

The builder for question answering translator.

Method Summary

Modifier and Type	Method	Description
CrossEncoderTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
CrossEncoderTranslator.Builder	optBatchifier(ai.djl.translate.Batchifier batchifier)	Sets the Batchifier for the Translator.
CrossEncoderTranslator.Builder	optIncludeTokenTypes(boolean includeTokenTypes)	Sets if include token types for the Translator.
CrossEncoderTranslator.Builder	optSigmoid(boolean sigmoid)	Sets if apply sigmoid 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 CrossEncoderTranslator.Builder optIncludeTokenTypes(boolean includeTokenTypes)

Sets if include token types for the Translator.

Parameters:

includeTokenTypes - true to include token types

Returns:

this builder

optSigmoid

public CrossEncoderTranslator.Builder optSigmoid(boolean sigmoid)

Sets if apply sigmoid for the Translator.

Parameters:

sigmoid - true to apply sigmoid

Returns:

this builder

optBatchifier

public CrossEncoderTranslator.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 CrossEncoderTranslator build()
                             throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs