matchzoo.datasets.toy package¶

Module contents¶

matchzoo.datasets.toy.load_data(stage='train', task='ranking', 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. return_classes (`bool`) – True to return classes for classification task, False otherwise.
Return type:	`Union`[`DataPack`, `Tuple`[`DataPack`, `list`]]
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()¶