Package ai.djl.mxnet.zoo.nlp.qa

Class MxBertQATranslator.Builder

java.lang.Object

ai.djl.modality.nlp.translator.QATranslator.BaseBuilder<MxBertQATranslator.Builder>

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

Enclosing class:

MxBertQATranslator

public static class MxBertQATranslator.Builder
extends ai.djl.modality.nlp.translator.QATranslator.BaseBuilder<MxBertQATranslator.Builder>

The builder for Bert QA translator.

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
protected MxBertQATranslator	build()	Builds the translator.
protected MxBertQATranslator.Builder	self()	Returns the builder.
MxBertQATranslator.Builder	setSeqLength(int seqLength)	Set the max length of the sequence to do the padding.

Methods inherited from class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder

configure, optBatchifier, optIncludeTokenTypes, optLocale, optMaxLabels, optMaxLength, optPadding, optTokenizer, optToLowerCase, optTruncation, optVocab

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

Builder

public Builder()

Method Details

setSeqLength

public MxBertQATranslator.Builder setSeqLength(int seqLength)

Set the max length of the sequence to do the padding.

Parameters:

seqLength - the length of the sequence

Returns:

builder

self

protected MxBertQATranslator.Builder self()

Returns the builder.

Specified by:

self in class ai.djl.modality.nlp.translator.QATranslator.BaseBuilder<MxBertQATranslator.Builder>

Returns:

the builder

build

protected MxBertQATranslator build()

Builds the translator.

Returns:

the new translator