from pathlib import Path
import sys

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

from model.gcrm_convectionnn import load_yaml, train_model

if __name__ == "__main__":
    config = load_yaml(ROOT / "conf/config.yaml")
    settings = {**config["model"], **config["training"], "format_version": config["project"]["format_version"]}
    history = train_model(str(ROOT / config["paths"]["data"]), str(ROOT / config["paths"]["checkpoint"]),
                          str(ROOT / config["paths"]["training_metrics"]), settings)
    print(f"loss={history['loss']}")