DatasetFactory

class paddle.fluid.dataset. DatasetFactory [源代码]

DatasetFactory是一个按数据集名称创建数据集的 "工厂"，可以创建“QueueDataset”，“InMemoryDataset”或“FileInstantDataset”，默认为“QueueDataset”。

代码示例

import paddle.fluid as fluid
dataset = fluid.DatasetFactory().create_dataset("InMemoryDataset")

方法

create_dataset(datafeed_class='QueueDataset')

创建“QueueDataset”，“InMemoryDataset” 或 “FileInstantDataset”，默认为“QueueDataset”。

参数

  • datafeed_class (str) – datafeed类名，为QueueDataset或InMemoryDataset。默认为QueueDataset。

代码示例

import paddle.fluid as fluid
dataset = fluid.DatasetFactory().create_dataset()