def train(
maps_dir: str,
train_dict: Dict[str, Any],
split_list: List[int],
erase_existing: bool = True,
):
maps_manager = MapsManager(maps_dir, train_dict, verbose="info")
maps_manager.train(split_list=split_list, overwrite=erase_existing)
def automatic_resume(model_path, user_split_list=None, verbose=0):
logger = getLogger("clinicadl")
verbose_list = ["warning", "info", "debug"]
maps_manager = MapsManager(model_path, verbose=verbose_list[verbose])
existing_split_list = maps_manager._find_splits()
stopped_splits = [
split
for split in existing_split_list
if "tmp"
in os.listdir(path.join(model_path, f"{maps_manager.split_name}-{split}"))
]
# Find finished split
finished_splits = list()
for split in existing_split_list:
if split not in stopped_splits:
performance_dir_list = [
performance_dir
for performance_dir in os.listdir(
path.join(model_path, f"{maps_manager.split_name}-{split}")
)
if "best-" in performance_dir
]
if len(performance_dir_list) > 0:
finished_splits.append(split)
split_manager = maps_manager._init_split_manager(split_list=user_split_list)
split_iterator = split_manager.split_iterator()
absent_splits = [
split
for split in split_iterator
if split not in finished_splits and split not in stopped_splits
]
# To ensure retro-compatibility with random search
logger.info(
f"Finished splits {finished_splits}\n"
f"Stopped splits {stopped_splits}\n"
f"Absent splits {absent_splits}"
)
if len(stopped_splits) > 0:
maps_manager.resume(stopped_splits)
if len(absent_splits) > 0:
maps_manager.train(absent_splits, overwrite=True)
def automatic_resume(model_path, verbose=0):
logger = getLogger("clinicadl")
verbose_list = ["warning", "info", "debug"]
maps_manager = MapsManager(model_path, verbose=verbose_list[verbose])
if len(glob(os.path.join(model_path, "fold-*"))) > 0:
raise ValueError(
"This MAPS cannot be resumed with the current version of ClinicaDL. "
"Please use the same version as for training or rename manually the folders "
"'fold-*' in 'split-*' to respect the new MAPS convention.")
split_list = sorted([
int(split.split("-")[1]) for split in os.listdir(model_path)
if split[:4:] == "split"
])
stopped_splits = [
split for split in split_list
if "tmp" in os.listdir(path.join(model_path, f"split-{split}"))
]
finished_splits = [
split for split in split_list if split not in stopped_splits
]
split_manager = maps_manager._init_split_manager()
split_iterator = split_manager.split_iterator()
absent_splits = [
split for split in split_iterator
if split not in finished_splits and split not in stopped_splits
]
logger.info(f"List of finished splits {finished_splits}")
logger.info(f"List of stopped splits {stopped_splits}")
logger.info(f"List of absent splits {absent_splits}")
# To ensure retro-compatibility with random search
logger.info(f"Finished splits {finished_splits}\n"
f"Stopped splits {stopped_splits}\n"
f"Absent splits {absent_splits}")
maps_manager.resume(stopped_splits)
maps_manager.train(absent_splits)
def train(maps_dir, train_dict, folds, erase_existing=True):
maps_manager = MapsManager(maps_dir, train_dict, verbose="info")
maps_manager.train(folds=folds, overwrite=erase_existing)