import argparse import pathlib if __name__ == "__main__": default_results_dir = 'results' default_models_dir = 'models' default_experiment_id = -1 parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument('--results_dir', nargs='?', type=str, default=default_results_dir, help='The output directory of the results.') parser.add_argument('--models_dir', nargs='?', type=str, default=default_models_dir, help='The output directory of the trained models.') parser.add_argument('--experiment_id', nargs='?', type=int, default=default_experiment_id, help='Compute the results of a single experiment id') args = parser.parse_args() pathlib.Path(args.results_dir).mkdir(parents=True, exist_ok=True) if args.experiment_id == -1: pass else: pass