cyto_dl.callbacks.outlier_detection module#

class cyto_dl.callbacks.outlier_detection.OutlierDetection(n_epochs, layer_names, save_dir)[source]#

Bases: Callback

calculate_mahalanobis_hook(layer_name)[source]#

flatten_activations(act)[source]#

on_load_checkpoint(trainer, pl_module, checkpoint)[source]#

on_predict_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)[source]#

on_predict_epoch_start(trainer, pl_module)[source]#

on_save_checkpoint(trainer, pl_module, checkpoint)[source]#

on_test_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)[source]#

on_test_epoch_start(trainer, pl_module)[source]#

on_train_epoch_start(trainer, pl_module)[source]#: Set forward hook.

update_covariance_hook(layer_name: str) → Callable[source]#