polyguard-rl/scripts/train_dose_model.py

#!/usr/bin/env python3
"""Train dose surrogate model placeholder."""

from __future__ import annotations

import json
import os
from pathlib import Path

from app.models.dosing.train_supervised import train_dosing_surrogate


def main() -> None:
    dataset_size = int(os.getenv("POLYGUARD_DOSE_DATASET_SIZE", "300"))
    result = train_dosing_surrogate(dataset_size=dataset_size)
    root = Path(__file__).resolve().parents[1]
    out = root / "outputs" / "reports"
    out.mkdir(parents=True, exist_ok=True)
    (out / "dose_train.json").write_text(json.dumps(result, ensure_ascii=True, indent=2), encoding="utf-8")
    print("dose_model_trained")


if __name__ == "__main__":
    main()