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', filter=False)

Load WikiQA data.

参数:
  • stage -- One of train, dev, and test.
  • task -- Could be one of ranking, classification or a matchzoo.engine.BaseTask instance.
  • filter -- Whether remove the questions without correct answers.
返回:

A DataPack if ranking, a tuple of (DataPack, classes) if classification.

Module contents