matchzoo.datasets.toy

Package Contents

matchzoo.datasets.toy.load_data(stage:str='train', task:str='ranking', return_classes:bool=False) → typing.Union[matchzoo.DataPack, typing.Tuple[matchzoo.DataPack, list]]

Load WikiQA data.

Parameters:
  • stage – One of train, dev, and test.
  • task – Could be one of ranking, classification or a matchzoo.engine.BaseTask instance.
  • 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.

Example

>>> import matchzoo as mz
>>> stages = 'train', 'dev', 'test'
>>> tasks = 'ranking', 'classification'
>>> for stage in stages:
...     for task in tasks:
...         _ = mz.datasets.toy.load_data(stage, task)
matchzoo.datasets.toy.load_embedding()