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 and Description
static class	QATranslator.BaseBuilder<T extends QATranslator.BaseBuilder> The builder for question answering translator.

Field Summary

Fields

Modifier and Type	Field and Description
protected Batchifier	batchifier
protected boolean	includeTokenTypes
protected java.util.Locale	locale
protected int	maxLabels
protected int	maxLength
protected boolean	padding
protected java.lang.String	tokenizerName
protected boolean	toLowerCase
protected boolean	truncation
protected java.lang.String	vocab

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	QATranslator(QATranslator.BaseBuilder<?> builder)

Method Summary

Modifier and Type	Method and Description
Batchifier	getBatchifier() Gets the Batchifier.

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

processInput

Methods inherited from interface ai.djl.translate.PostProcessor

processOutput

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()

Gets the Batchifier.

Specified by:

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

Returns:

the Batchifier