OneScience-Group/AtmosphericDA-DModel

1from pathlib import Path2import sys3 4ROOT = Path(__file__).resolve().parents[1]5sys.path.insert(0, str(ROOT))6 7from model.atmosphericda_dmodel import load_yaml, run_inference8 9 10if __name__ == "__main__":11    config = load_yaml(ROOT / "conf/config.yaml")12    metrics = run_inference(13        str(ROOT / config["paths"]["data"]),14        str(ROOT / config["paths"]["checkpoint"]),15        str(ROOT / config["paths"]["predictions"]), config["project"]["format_version"],16    )17    print(f"saved predictions: {metrics['hybrid_prediction'].shape}")18