morphoclass.training.transfer_learning module

Implementation of the transfer learning curves command.

morphoclass.training.transfer_learning.transfer_learning_curves(input_csv, features_dir, image_path, checkpoints_directory, model_class, model_params, dataset_name, feature_extractor_name, optimizer_class, optimizer_params, n_epochs, batch_size, seed, checkpoint_path_pretrained, frozen_backbone=False)

Generate TL curves by iterating through splits of different sizes.