Skip to content

metalists

metalists

Classes

Functions

get_augmentation_list

get_augmentation_list(max_augmentators=5)

Includes ID and all augmentations when used with genericdatagenerator

Source code in niceml/experiments/metalists.py
def get_augmentation_list(max_augmentators: int = 5) -> List[MetaFunction]:
    """Includes ID and all augmentations when used with genericdatagenerator"""
    meta_functions = [ExperimentIdExtraction()]
    for idx in range(max_augmentators):
        cur_aug_conf_path = AUGMENTATIONS_CONFIG_PATH + [idx]
        meta_functions.append(
            ConfigInfoExtractor(
                f"augmentation_{idx}",
                cur_aug_conf_path,
                info_format_func=hydra_instance_format,
            )
        )
    return meta_functions