deepr.readers.GeneratorReader

class deepr.readers.GeneratorReader(generator_fn, output_types, output_shapes=None)[source]

Reader Class for datasets using generator functions

generator_fn

Generator function, yields features, labels

Type:

Callable

output_types

Generator outputs dtypes

Type:

Nested structure of tf.DType

output_shapes

Generator outputs shapes

Type:

Nested structure of tf.TensorShape

__init__(generator_fn, output_types, output_shapes=None)[source]

Methods

__init__(generator_fn, output_types[, ...])

as_dataset()

Build a tf.data.Dataset