CometCallback

class radionets.core.CometCallback(name, validation_data, plot_n_epochs, amp_phase, scale)

Bases: Callback

Callback for logging training metrics and visualizations to Comet ML.

This callback logs training and validation losses, and creates plots for predictions and Fourier-transformed data for monitoring during training.

Parameters:

namestr

Project name for the Comet ML experiment.

validation_datastr or Path

Path to the validation dataset.

plot_n_epochsint

Frequency of plotting (every n epochs).

amp_phasebool

Whether to use amplitude-phase representation.

scalestr

Scaling method for data.

Methods Summary

after_epoch()
after_train()
after_validate()
plot_test_fft()
plot_test_pred()

Methods Documentation

after_epoch()

after_train()

after_validate()

plot_test_fft()

plot_test_pred()