chainer.datasets.PickleDatasetWriter

class chainer.datasets.PickleDatasetWriter(writer, protocol=4)

Writer class that makes PickleDataset.

To make PickleDataset, a user needs to prepare data using PickleDatasetWriter.

Parameters

• writer – File like object that supports write and tell methods.

• protocol (int) – Valid protocol for pickle.

Methods

__enter__()

__exit__(exc_type, exc_value, traceback)

close()

flush()

write(x)