p2pfl.learning.frameworks.tensorflow.callbacks.keras_logger module¶

Keras Logger for P2PFL.

class p2pfl.learning.frameworks.tensorflow.callbacks.keras_logger.FederatedLogger(node_name)[source]¶

Bases: Callback, P2PFLCallback

Keras Logger for Federated Learning. Handles local training logging.

Parameters:: node_name (str) – Name of the node.

static get_name()[source]¶

Get the name of the callback.

Return type:: str

on_epoch_end(epoch, logs=None)[source]¶: Log metrics at the end of each epoch.

on_train_batch_end(batch, logs=None)[source]¶: Log metrics at the end of each batch (optional).