ai.djl.mxnet.zoo.nlp.qa

Class BertQATranslator

java.lang.Object

ai.djl.mxnet.zoo.nlp.qa.BertQATranslator

All Implemented Interfaces:

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

public class BertQATranslator
extends java.lang.Object
implements ai.djl.translate.Translator<QAInput,java.lang.String>

The translator for BertQAModelLoader.

Method Summary

Modifier and Type	Method and Description
ai.djl.translate.Batchifier	getBatchifier()
ai.djl.ndarray.NDList	processInput(ai.djl.translate.TranslatorContext ctx, 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

getBatchifier

public ai.djl.translate.Batchifier getBatchifier()

Specified by:

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

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
                                          QAInput input)
                                   throws java.io.IOException

Specified by:

processInput in interface ai.djl.translate.PreProcessor<QAInput>

Throws:

java.io.IOException

processOutput

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

Specified by:

processOutput in interface ai.djl.translate.PostProcessor<java.lang.String>