bird_cloud_gnn.callback

Module Contents

Classes

TensorboardCallback	Callback to populate Tensorboard.
EarlyStopperCallback	Callback to check early stopping.
CombinedCallback	Helper to combine multiple callbacks.

class bird_cloud_gnn.callback.TensorboardCallback(**kwargs)

Callback to populate Tensorboard.

This class provides a callback function to populate Tensorboard with scalar and histogram summaries during training. The callback function takes in epoch values and adds scalar and histogram summaries to Tensorboard for each field in the epoch values that matches certain criteria.

Parameters:

**kwargs – Additional arguments to pass to the SummaryWriter constructor.

writer

A SummaryWriter object used to write summaries to Tensorboard.

__call__(epoch_values)

Callback function to populate Tensorboard with scalar and histogram summaries.

Parameters:

epoch_values – A dictionary containing the values for each field at the current epoch.

Returns:

False, indicating that the training should continue.

class bird_cloud_gnn.callback.EarlyStopperCallback(**kwargs)

Callback to check early stopping.

This callback is used to check if the training should be stopped early based on the validation loss.

__call__(epoch_values)

class bird_cloud_gnn.callback.CombinedCallback(callbacks)

Helper to combine multiple callbacks.

This class allows multiple callbacks to be combined into a single callback. The callbacks are called in the given sequence and if one of them returns True, the subsequent callbacks are not called.

Parameters:

callbacks (iterable) – List of callbacks to be combined.

Returns:

True if any of the callbacks return True, False otherwise.

Return type:

bool

__call__(epoch_values)