deepblink.datasets.sequence module

SequenceDataset class.

class deepblink.datasets.sequence.SequenceDataset(x: numpy.ndarray, y: numpy.ndarray, batch_size: int = 16, augment_fn: Callable = None, format_fn: Callable = None, overfit: bool = False)

Bases: keras.utils.data_utils.Sequence

Custom Sequence class used to feed data into model.fit.

Parameters:

• x_list – List of inputs.
• y_list – List of targets.
• batch_size – Size of one mini-batch.
• augment_fn – Function to augment one mini-batch of x and y.
• format_fn – Function to format raw data to model input.
• overfit – If only one batch should be used thereby causing overfitting.

on_epoch_end() → None

Shuffle data after every epoch.