Package ai.djl.modality.nlp.translator
Class QATranslator
- java.lang.Object
-
- ai.djl.modality.nlp.translator.QATranslator
-
- All Implemented Interfaces:
PostProcessor<java.lang.String>,PreProcessor<QAInput>,Translator<QAInput,java.lang.String>
public abstract class QATranslator extends java.lang.Object implements Translator<QAInput,java.lang.String>
An abstract class to define the question answering translator.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classQATranslator.BaseBuilder<T extends QATranslator.BaseBuilder>The builder for question answering translator.
-
Field Summary
Fields Modifier and Type Field Description protected Batchifierbatchifierprotected booleanincludeTokenTypesprotected java.util.Localelocaleprotected intmaxLabelsprotected intmaxLengthprotected booleanpaddingprotected java.lang.StringtokenizerNameprotected booleantoLowerCaseprotected booleantruncationprotected java.lang.Stringvocab
-
Constructor Summary
Constructors Modifier Constructor Description protectedQATranslator(QATranslator.BaseBuilder<?> builder)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description BatchifiergetBatchifier()Returns theBatchifier.-
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.PostProcessor
processOutput
-
Methods inherited from interface ai.djl.translate.PreProcessor
processInput
-
Methods inherited from interface ai.djl.translate.Translator
getExpansions, prepare, toBatchTranslator, toBatchTranslator
-
-
-
-
Field Detail
-
batchifier
protected Batchifier batchifier
-
tokenizerName
protected java.lang.String tokenizerName
-
vocab
protected java.lang.String vocab
-
locale
protected java.util.Locale locale
-
toLowerCase
protected boolean toLowerCase
-
includeTokenTypes
protected boolean includeTokenTypes
-
padding
protected boolean padding
-
truncation
protected boolean truncation
-
maxLength
protected int maxLength
-
maxLabels
protected int maxLabels
-
-
Constructor Detail
-
QATranslator
protected QATranslator(QATranslator.BaseBuilder<?> builder)
-
-
Method Detail
-
getBatchifier
public Batchifier getBatchifier()
Returns theBatchifier.- Specified by:
getBatchifierin interfaceTranslator<QAInput,java.lang.String>- Returns:
- the
Batchifier
-
-