qumphy.callbacks.pulsedb module

File: qumphy/models/lightning_callbacks.py Project: 22HLT01 QUMPHY Contact: oskar.pfeffer@ptb.de Gitlab: https://gitlab.com/qumphy Description: Pytorch lightning callbacks.

class qumphy.callbacks.pulsedb.PulseDBLogging(*args: Any, **kwargs: Any)

Bases: BaseLoggingCallback

Pytorch lightning callback for logging for the PulseDB dataset.

Initialize with a list of quantities to log and a flag to specify which pressure to log. Loss is always logged.

Available quantities:

• mae

• rmse

• std

log_epoch_end(trainer, pl_module, stage)

Override this method to log the metrics at the end of the epoch.

log_ieee_metrics(pl_module, values, stage)

log_mae(pl_module, values, stage)

log_rmse(pl_module, values, stage)

log_std(pl_module, values, stage)

set_function_dictionary()

class qumphy.callbacks.pulsedb.PulseDBLogging_Ensemble(*args: Any, **kwargs: Any)

Bases: PulseDBLogging

on_test_epoch_end(trainer, pl_module)

class qumphy.callbacks.pulsedb.PulseDBLogging_MCD(*args: Any, **kwargs: Any)

Bases: PulseDBLogging

on_test_epoch_end(trainer, pl_module)

class qumphy.callbacks.pulsedb.PulseDBLogging_Pinballloss(*args: Any, **kwargs: Any)

Bases: BaseLoggingCallback

on_test_epoch_end(trainer, pl_module)