tflite_support.task.text.BertQuestionAnswerer

Class that performs Bert question answering on text.

options

Methods

answer

View source

Answers question based on the context.

Could be empty if no answer was found from the given context.

Args
context Context the question bases on.
question Question to ask.

Returns
Question answerer result.

Raises
ValueError If any of the input arguments is invalid.
RuntimeError If failed to calculate the embedding vector.

create_from_file

View source

Creates the BertQuestionAnswerer object from a TensorFlow Lite model.

Args
file_path Path to the model.

Returns
BertQuestionAnswerer object that's created from the model file.

Raises
ValueError If failed to create BertQuestionAnswerer object from the provided file such as invalid file.
RuntimeError If other types of error occurred.

create_from_options

View source

Creates the BertQuestionAnswerer object from the options.

Args
options Options for the Bert question answerer task.

Returns
BertQuestionAnswerer object that's created from options.

Raises
ValueError If failed to create BertQuestionAnswerer object from BertQuestionAnswererOptions such as missing the model.
RuntimeError If other types of error occurred.