Combines consecutive elements of this dataset into batches.

Combines consecutive elements of this dataset into batches.

dataset_batch(dataset, batch_size, drop_remainder = FALSE)

Arguments

dataset

A dataset

batch_size

An integer, representing the number of consecutive elements of this dataset to combine in a single batch.

drop_remainder

Ensure that yoru batches have a fixed size by omiting any final smaller batch if it's present. Note that this is required for use with the Keras tensor inputs to fit/evaluate/etc.

Value

A dataset

See also

Other dataset methods: dataset_cache, dataset_concatenate, dataset_decode_delim, dataset_filter, dataset_interleave, dataset_map, dataset_padded_batch, dataset_prefetch, dataset_repeat, dataset_shuffle, dataset_skip, dataset_take