30 lines
753 B
Python
30 lines
753 B
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(PROJECT_ROOT / "src"))
|
|
|
|
from ml_crypto_lab.core.config import load_yaml
|
|
from ml_crypto_lab.train.runner import run_full_experiment
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--config", required=True, help="Path to experiment yaml")
|
|
args = parser.parse_args()
|
|
cfg = load_yaml(args.config)
|
|
result = run_full_experiment(cfg)
|
|
print("=" * 120)
|
|
print("DONE")
|
|
print("run_id:", result["run_id"])
|
|
print("run_dir:", result["run_dir"])
|
|
print("best rows:")
|
|
print(result["summary"].head(20))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|