ai.djl.mxnet.zoo.nlp.qa

Class QAInput

java.lang.Object

ai.djl.mxnet.zoo.nlp.qa.QAInput

public class QAInput
extends java.lang.Object

The input container for BERT QA model.

Constructor Summary

Constructors

Constructor and Description
QAInput(java.lang.String question, java.lang.String paragraph, int seqLength) Creates the BERT QA model.

Method Summary

Modifier and Type	Method and Description
java.lang.String	getParagraph() Gets the resource document that contains the answer.
java.lang.String	getQuestion() Gets the question for the model.
int	getSeqLength() Gets the sequence length.

Methods inherited from class java.lang.Object

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

Constructor Detail

QAInput

public QAInput(java.lang.String question,
               java.lang.String paragraph,
               int seqLength)

Creates the BERT QA model.

Parameters:

question - the question for the model

paragraph - the resource document that contains the answer

seqLength - the maximum length of the sentence

Method Detail

getQuestion

public java.lang.String getQuestion()

Gets the question for the model.

Returns:

the question for the model

getParagraph

public java.lang.String getParagraph()

Gets the resource document that contains the answer.

Returns:

the resource document that contains the answer

getSeqLength

public int getSeqLength()

Gets the sequence length.

Returns:

the sequence length