ai.djl.modality.nlp.translator

Class Seq2SeqTranslator

java.lang.Object

ai.djl.modality.nlp.translator.Seq2SeqTranslator

All Implemented Interfaces:

PostProcessor<java.lang.String>, PreProcessor<java.lang.String>, Translator<java.lang.String,java.lang.String>

public class Seq2SeqTranslator
extends java.lang.Object
implements Translator<java.lang.String,java.lang.String>

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

Constructor Summary

Constructors

Constructor and Description
Seq2SeqTranslator()

Method Summary

Modifier and Type	Method and Description
Batchifier	getBatchifier() Gets the Batchifier.
NDList	processInput(TranslatorContext ctx, java.lang.String input) Processes the input and converts it to NDList.
java.lang.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

prepare

Methods inherited from interface ai.djl.translate.PreProcessor

getPipeline

Constructor Detail

Seq2SeqTranslator

public Seq2SeqTranslator()

Method Detail

processOutput

public java.lang.String processOutput(TranslatorContext ctx,
                                      NDList list)

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<java.lang.String>

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

processInput

public NDList processInput(TranslatorContext ctx,
                           java.lang.String input)

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<java.lang.String>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing

getBatchifier

public Batchifier getBatchifier()

Gets the Batchifier.

Specified by:

getBatchifier in interface Translator<java.lang.String,java.lang.String>

Returns:

the Batchifier