matchzoo.dataloader.dataset_builder

Module Contents

class matchzoo.dataloader.dataset_builder.DatasetBuilder(**kwargs)

Bases: object

Dataset Bulider. In essense a wrapped partial function.

Example

>>> import matchzoo as mz
>>> builder = mz.dataloader.DatasetBuilder(
...     mode='point'
... )
>>> data = mz.datasets.toy.load_data()
>>> gen = builder.build(data)
>>> type(gen)
<class 'matchzoo.dataloader.dataset.Dataset'>

build(self, data_pack, **kwargs)

Build a Dataset.

Parameters:

• data_pack – DataPack to build upon.
• kwargs – Additional keyword arguments to override the keyword arguments passed in __init__.