epitome.generators.generator_to_tf_dataset

epitome.generators.generator_to_tf_dataset(g, batch_size, shuffle_size, prefetch_size)

Generates a tensorflow dataset from a data generator.

Parameters

• g – data generator

• batch_size – number of elements in generator to combine into a single batch

• shuffle_size – number of elements from the generator fromw which the new dataset will shuffle

• prefetch_size – maximum number of elements that will be buffered when prefetching

Returns

tuple of (label shape, tf.data.Dataset)