cellflow.training.ComputationCallback

class cellflow.training.ComputationCallback

Base class for computation callbacks in the CellFlowTrainer

Methods

on_log_iteration(valid_source_data, ...)	Called at each validation/log iteration to compute metrics
on_train_begin()	Called at the beginning of training to initiate metric computation
on_train_end(valid_source_data, ...)	Called at the end of training to compute metrics