scripts/train_risk_model.py

#!/usr/bin/env python3
"""Train tabular risk model placeholder."""

from __future__ import annotations

import json
import os
from pathlib import Path

from app.models.tabular.train import train_tabular_model


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


if __name__ == "__main__":
    main()