

ValidationCallback

class ValidationCallback

Callback for validation loss.

__init__

def __init__(summarizer, val_dataset, loss_function, batch_size)

Initializes the Callback.

Args

• summarizer: Summarizer to validate.

• val_dataset: Validation dataset to validate the model on.

• loss_function: Loss function to apply to calculate the validation score.

• batch_size: Batch size of the validation dataset, needed for initializing the model.

on_epoch_end

def on_epoch_end(batch, logs)