primeqa.tableqa.models.tableqa_model.TableQAModel

class primeqa.tableqa.models.tableqa_model.TableQAModel(model_name_path, config=None)

Bases: object

Methods

predict

This function takes a table dictionary and a list of queries as input and returns the answer to the queries using the TableQA model.

Attributes

model	Propery of TableQA model.
tokenizer	Property of TableQG model.

property model

Propery of TableQA model. :returns: Sequence to sequence model object (based on model name)

predict(data_dict, queries_list)

This function takes a table dictionary and a list of queries as input and returns the answer to the queries using the TableQA model.

Parameters

• data_dict (Dict) – Table in dict format

• queries_list (List) – List of queries

Returns

Returns a dictionary of query and the predicted answer.

Return type

Dict

property tokenizer

Property of TableQG model. :returns: Tokenizer class object based on the model name/ path