Package ai.djl.pytorch.zoo.nlp.qa
Class PtBertQATranslator
- java.lang.Object
-
- ai.djl.modality.nlp.translator.QATranslator
-
- ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator
-
- 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 PtBertQATranslator extends ai.djl.modality.nlp.translator.QATranslator
TheTranslator
for PyTorch Question Answering model.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
PtBertQATranslator.Builder
The builder for Bert QA translator.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static PtBertQATranslator.Builder
builder()
Creates a builder to build aPtBertQATranslator
.static PtBertQATranslator.Builder
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aPtSSDTranslatorBuilder
with specified arguments.void
prepare(ai.djl.translate.TranslatorContext ctx)
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)
-
-
-
Method Detail
-
prepare
public void prepare(ai.djl.translate.TranslatorContext ctx) throws java.io.IOException
- Throws:
java.io.IOException
-
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 PtBertQATranslator.Builder builder()
Creates a builder to build aPtBertQATranslator
.- Returns:
- a new builder
-
builder
public static PtBertQATranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aPtSSDTranslatorBuilder
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-
-