Package ai.djl.huggingface.translator

Class SparseRetrievalTranslator.Builder

java.lang.Object

ai.djl.huggingface.translator.SparseRetrievalTranslator.Builder

Enclosing class:

SparseRetrievalTranslator

public static final class SparseRetrievalTranslator.Builder
extends Object

The builder for question answering translator.

Method Summary

Modifier and Type	Method	Description
SparseRetrievalTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
SparseRetrievalTranslator.Builder	optReturnDenseEmbedding(boolean returnDenseEmbedding)	Sets if apply sigmoid for the Translator.
SparseRetrievalTranslator.Builder	optSparseLinear(String sparseLinear)	Sets the sparse linear layer model file for the Translator.

Methods inherited from class java.lang.Object

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

Method Details

optReturnDenseEmbedding

public SparseRetrievalTranslator.Builder optReturnDenseEmbedding(boolean returnDenseEmbedding)

Sets if apply sigmoid for the Translator.

Parameters:

returnDenseEmbedding - true to output dense embedding

Returns:

this builder

optSparseLinear

public SparseRetrievalTranslator.Builder optSparseLinear(String sparseLinear)

Sets the sparse linear layer model file for the Translator.

Parameters:

sparseLinear - path to sparse linear layer model file

Returns:

this builder

configure

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

Configures the builder with the model arguments.

Parameters:

arguments - the model arguments

build

public SparseRetrievalTranslator build()
                                throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs