scripts/inference.py

from pathlib import Path
import sys

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

from model.pbl_emulator import load_yaml, run_inference

if __name__ == "__main__":
    config = load_yaml(ROOT / "conf/config.yaml")
    summary = run_inference(ROOT / config["paths"]["data"], ROOT / config["paths"]["checkpoint"], ROOT / config["paths"]["predictions"])
    print(summary)