| 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) | |