Spaces:
Running
Running
File size: 672 Bytes
877add7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | #!/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()
|