| import argparse |
| import subprocess |
| import sys |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parents[1] |
| TASK_MAP = { |
| 'synthetic': [sys.executable, str(ROOT / 'scripts' / 'run_synthetic.py')], |
| 'cifar10': [sys.executable, str(ROOT / 'scripts' / 'run_softmax.py')], |
| 'topics': [sys.executable, str(ROOT / 'scripts' / 'run_topics.py')], |
| 'samson': [sys.executable, str(ROOT / 'scripts' / 'run_hyperspectral.py')], |
| 'pbmc': [sys.executable, str(ROOT / 'scripts' / 'run_bulk_deconv.py')], |
| 'utkface': [sys.executable, str(ROOT / 'scripts' / 'run_age_ldl.py')], |
| 'affectivetext': [sys.executable, str(ROOT / 'scripts' / 'run_affective_text.py')], |
| } |
| parser = argparse.ArgumentParser() |
| parser.add_argument('task', choices=TASK_MAP) |
| parser.add_argument('extra', nargs=argparse.REMAINDER) |
| args = parser.parse_args() |
| subprocess.check_call(TASK_MAP[args.task] + args.extra) |
|
|
