File size: 1,126 Bytes
7affe31 | 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 26 27 28 29 30 31 32 33 34 35 | from __future__ import annotations
import argparse
from pathlib import Path
from timeout_risk_model import load_model, predict_timeout_probability
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Predict timeout probability for one order time.")
parser.add_argument("order_time", help="Order time, for example: 2026-05-24 18:30")
parser.add_argument("--product-name", default="", help="Optional product name.")
parser.add_argument(
"--model",
type=Path,
default=Path(__file__).resolve().parent / "timeout_model.json",
help="Path to timeout_model.json.",
)
return parser.parse_args()
def main() -> None:
args = parse_args()
model = load_model(args.model)
result = predict_timeout_probability(model, args.order_time, args.product_name)
print(f"order_time: {result['order_time']}")
print(f"probability: {result['probability']:.1%}")
print(f"risk_level: {result['risk_level']}")
print(f"mode: {result['mode']}")
print(f"calibration: {result['calibration']}")
if __name__ == "__main__":
main()
|