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()