TrainingΒΆ

training.BaseCallback()

Base class for callbacks in the CellFlowTrainer

training.CallbackRunner(callbacks)

Runs a set of computational and logging callbacks in the CellFlowTrainer

training.ComputationCallback()

Base class for computation callbacks in the CellFlowTrainer

training.LoggingCallback()

Base class for logging callbacks in the CellFlowTrainer

training.Metrics(metrics[, metric_aggregations])

Callback to compute metrics on validation data during training

training.PCADecodedMetrics(ref_adata, metrics)

Callback to compute metrics on decoded validation data during training

training.VAEDecodedMetrics(vae, adata, metrics)

Callback to compute metrics on decoded validation data during training

training.WandbLogger(project, out_dir, ...)

Callback to log data to Weights and Biases

training.CellFlowTrainer(solver[, ...])

Trainer for the OTFM/GENOT solver with a conditional velocity field.