Package ai.djl.modality.nlp.translator

Class SimpleText2TextTranslator

java.lang.Object

ai.djl.modality.nlp.translator.SimpleText2TextTranslator

All Implemented Interfaces:

PostProcessor<String>, PreProcessor<String>, Translator<String,String>

public class SimpleText2TextTranslator
extends Object
implements Translator<String,String>

A Translator that performs pre-process and post-processing for a sequence-to-sequence text model.

Constructor Summary

Constructors

Constructor	Description
SimpleText2TextTranslator()

Method Summary

Modifier and Type	Method	Description
Batchifier	getBatchifier()	Returns the Batchifier.
NDList	processInput(TranslatorContext ctx, String input)	Processes the input and converts it to NDList.
String	processOutput(TranslatorContext ctx, NDList list)	Processes the output NDList to the corresponding output object.

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

batchProcessInput, batchProcessOutput, getExpansions, prepare

Constructor Details

SimpleText2TextTranslator

public SimpleText2TextTranslator()

Method Details

processOutput

public String processOutput(TranslatorContext ctx,
 NDList list)

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<String>

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference, usually immutable in engines like PyTorch. @see Issue 1774

Returns:

the output object of expected type

processInput

public NDList processInput(TranslatorContext ctx,
 String input)

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<String>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing

getBatchifier

public Batchifier getBatchifier()

Returns the Batchifier.

Specified by:

getBatchifier in interface Translator<String,String>

Returns:

the Batchifier