ai.djl.mxnet.zoo.nlp.qa

Class MxBertQATranslator

java.lang.Object

ai.djl.modality.nlp.translator.QATranslator

ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

All Implemented Interfaces:

ai.djl.translate.PostProcessor<java.lang.String>, ai.djl.translate.PreProcessor<ai.djl.modality.nlp.qa.QAInput>, ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,java.lang.String>

public class MxBertQATranslator
extends ai.djl.modality.nlp.translator.QATranslator

The translator for BertQAModelLoader.

See Also:

BertQAModelLoader

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MxBertQATranslator.Builder The builder for Bert QA translator.

Nested classes/interfaces inherited from class ai.djl.modality.nlp.translator.QATranslator

ai.djl.modality.nlp.translator.QATranslator.BaseBuilder<T extends ai.djl.modality.nlp.translator.QATranslator.BaseBuilder>

Field Summary

Fields inherited from class ai.djl.modality.nlp.translator.QATranslator

batchifier

Method Summary

Modifier and Type	Method and Description
static MxBertQATranslator.Builder	builder() Creates a builder to build a MxBertQATranslator.
ai.djl.translate.Batchifier	getBatchifier()
void	prepare(ai.djl.ndarray.NDManager manager, ai.djl.Model model)
ai.djl.ndarray.NDList	processInput(ai.djl.translate.TranslatorContext ctx, ai.djl.modality.nlp.qa.QAInput input)
java.lang.String	processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)

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.PreProcessor

getPipeline

Method Detail

prepare

public void prepare(ai.djl.ndarray.NDManager manager,
                    ai.djl.Model model)
             throws java.io.IOException

Throws:

java.io.IOException

getBatchifier

public ai.djl.translate.Batchifier getBatchifier()

Specified by:

getBatchifier in interface ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,java.lang.String>

Overrides:

getBatchifier in class ai.djl.modality.nlp.translator.QATranslator

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
                                          ai.djl.modality.nlp.qa.QAInput input)

processOutput

public java.lang.String processOutput(ai.djl.translate.TranslatorContext ctx,
                                      ai.djl.ndarray.NDList list)

builder

public static MxBertQATranslator.Builder builder()

Creates a builder to build a MxBertQATranslator.

Returns:

a new builder