matchzoo.datasets.wiki_qa package

Submodules

matchzoo.datasets.wiki_qa.load_data module

WikiQA data loader.

matchzoo.datasets.wiki_qa.load_data.load_data(stage='train', task='ranking', filtered=False, return_classes=False)

Load WikiQA data.

Parameters:
  • stage (str) – One of train, dev, and test.
  • task (str) – Could be one of ranking, classification or a matchzoo.engine.BaseTask instance.
  • filtered (bool) – Whether remove the questions without correct answers.
  • return_classes (bool) – True to return classes for classification task, False otherwise.
Return type:

Union[DataPack, tuple]
Returns:

A DataPack unless task is classificiation and return_classes is True: a tuple of (DataPack, classes) in that case.

Module contents