File size: 522 Bytes
702a3cf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | 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, train_model
if __name__ == "__main__":
config = load_yaml(ROOT / "conf/config.yaml")
settings = {**config["model"], **config["training"], "format_version": config["project"]["format_version"]}
result = train_model(ROOT / config["paths"]["data"], ROOT / config["paths"]["checkpoint"], ROOT / config["paths"]["training_metrics"], settings)
print(result)
|