|View source on GitHub|
APIs to train a model that can answer questions based on a predefined text.
class BertQaSpec: A specification of BERT model for question answering.
class DataLoader: DataLoader for question answering.
class QuestionAnswer: QuestionAnswer class for inference and exporting to tflite.
MobileBertQaSpec(...): Creates MobileBert model spec for the question answer task. See also:
MobileBertQaSquadSpec(...): Creates MobileBert model spec that's already retrained on SQuAD1.1 for the question answer task. See also:
create(...): Loads data and train the model for question answer.