Package ai.djl.huggingface.translator

Class FillMaskTranslator

java.lang.Object

ai.djl.huggingface.translator.FillMaskTranslator

All Implemented Interfaces:

ai.djl.translate.PostProcessor<ai.djl.modality.Classifications>, ai.djl.translate.PreProcessor<java.lang.String>, ai.djl.translate.Translator<java.lang.String,ai.djl.modality.Classifications>

public class FillMaskTranslator
extends java.lang.Object
implements ai.djl.translate.Translator<java.lang.String,ai.djl.modality.Classifications>

The translator for Huggingface fill mask model.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	FillMaskTranslator.Builder	The builder for fill mask translator.

Method Summary

Modifier and Type	Method	Description
static FillMaskTranslator.Builder	builder(HuggingFaceTokenizer tokenizer)	Creates a builder to build a FillMaskTranslator.
static FillMaskTranslator.Builder	builder(HuggingFaceTokenizer tokenizer, java.util.Map<java.lang.String,?> arguments)	Creates a builder to build a FillMaskTranslator.
ai.djl.translate.Batchifier	getBatchifier()
ai.djl.ndarray.NDList	processInput(ai.djl.translate.TranslatorContext ctx, java.lang.String input)
ai.djl.modality.Classifications	processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
FillMaskBatchTranslator	toBatchTranslator(ai.djl.translate.Batchifier batchifier)

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.translate.Translator

getExpansions, prepare, toBatchTranslator

Method Detail

getBatchifier

public ai.djl.translate.Batchifier getBatchifier()

Specified by:

getBatchifier in interface ai.djl.translate.Translator<java.lang.String,ai.djl.modality.Classifications>

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
                                          java.lang.String input)
                                   throws ai.djl.translate.TranslateException

Specified by:

processInput in interface ai.djl.translate.PreProcessor<java.lang.String>

Throws:

ai.djl.translate.TranslateException

processOutput

public ai.djl.modality.Classifications processOutput(ai.djl.translate.TranslatorContext ctx,
                                                     ai.djl.ndarray.NDList list)

Specified by:

processOutput in interface ai.djl.translate.PostProcessor<ai.djl.modality.Classifications>

toBatchTranslator

public FillMaskBatchTranslator toBatchTranslator(ai.djl.translate.Batchifier batchifier)

Specified by:

toBatchTranslator in interface ai.djl.translate.Translator<java.lang.String,ai.djl.modality.Classifications>

builder

public static FillMaskTranslator.Builder builder(HuggingFaceTokenizer tokenizer)

Creates a builder to build a FillMaskTranslator.

Parameters:

tokenizer - the tokenizer

Returns:

a new builder

builder

public static FillMaskTranslator.Builder builder(HuggingFaceTokenizer tokenizer,
                                                 java.util.Map<java.lang.String,?> arguments)

Creates a builder to build a FillMaskTranslator.

Parameters:

tokenizer - the tokenizer

arguments - the models' arguments

Returns:

a new builder