matchzoo.datasets.wiki_qa

Submodules

• matchzoo.datasets.wiki_qa.load_data

Package Contents

matchzoo.datasets.wiki_qa.load_data(stage: str = 'train', task: typing.Union[str, BaseTask] = 'ranking', filtered: bool = False, return_classes: bool = False) → typing.Union[matchzoo.DataPack, tuple]

Load WikiQA data.

Parameters:

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

Returns:

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