mednet.utils.tensorboard

Functions

scalars_to_dict(logdir)

Return scalars stored in tensorboard event files.

mednet.utils.tensorboard.scalars_to_dict(logdir)[source]

Return scalars stored in tensorboard event files.

This method will gather all tensorboard event files produced by a training run, and will return a dictionary with all collected scalars, ready for plotting.

Parameters:

logdir (Path) – Directory containing the event files.

Returns:

A dictionary where keys represent all scalar names, and values correspond to a tuple that contains an array with epoch numbers (when values were taken), when the monitored values themselves. The lists are pre-sorted by epoch number.

Return type:

dict[str, tuple[list[int], list[float]]]