run_experiments.py

import argparse

from data.main import main as data_main
from training.main import main as training_main
from xai.main import main as xai_main
from evaluation.main import main as evaluation_main
from visualization.main import main as visualization_main
from bias.main import main as bias_main
from bias.model_bias import main as model_bias_main
from training.sweep import main as sweep_main
from utils import load_json_file


def get_command_line_arguments() -> argparse.Namespace:
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--config',
        dest='config',
        required=True,
        help='File path to project config.',
        type=str,
        default=1,
    )
    parser.add_argument(
        '--mode',
        dest='mode',
        required=True,
        help='Modes: training, xai, evaluation, visualization, bias, sweep.',
        type=str,
        default=1,
    )
    return parser.parse_args()


def main(config_path: str, mode: str):
    project_config = load_json_file(file_path=config_path)
    Modes[mode](config=project_config)


Modes = {
    'training': training_main,
    'xai': xai_main,
    'evaluation': evaluation_main,
    'visualization': visualization_main,
    'bias': bias_main,
    'model_bias': model_bias_main,
    'sweep': sweep_main,
}

if __name__ == '__main__':
    args = get_command_line_arguments()
    main(config_path=args.config, mode=args.mode)