matchzoo.datasets.toy package

Module contents

matchzoo.datasets.toy.load_data(stage='train', task='ranking')

Load WikiQA data.

参数:

• stage -- One of train, dev, and test.
• task -- Could be one of ranking, classification or a matchzoo.engine.BaseTask instance.

返回:

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

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)