openfl.callbacks.MetricWriter#
- class openfl.callbacks.MetricWriter(log_dir='./logs/', use_tensorboard=True)[source]#
Log scalar metrics at the end of each round.
- Parameters:
log_dir (str)
use_tensorboard (bool)
- log_dir#
Path to write logs as lines of JSON. Defaults to ./logs.
- use_tensorboard#
If True, writes scalar summaries to TensorBoard under log_dir.
- __init__(log_dir='./logs/', use_tensorboard=True)[source]#
- Parameters:
log_dir (str)
use_tensorboard (bool)
Methods
__init__([log_dir, use_tensorboard])on_experiment_begin([logs])Open file handles for logging.
on_experiment_end([logs])Cleanup.
on_round_begin(round_num[, logs])Callback function to be executed at the beginning of a round.
on_round_end(round_num[, logs])Log metrics.
set_params(params)set_tensor_db(tensor_db)