demo.py

import json
from pathlib import Path

from src.inference_engine import LatencyDqnOnnxAgent
from src.state_serializer import LatencyStateSerializer
from src.action_interpreter import ActionInterpreter

MODEL_PATH = "model/5g_latency_opt_dqn_model.onnx"
CONFIG_PATH = "model/model_config.json"
INPUT_JSON = "validation241204.json"


def main():
    agent = LatencyDqnOnnxAgent(MODEL_PATH, CONFIG_PATH)
    serializer = LatencyStateSerializer(CONFIG_PATH)
    interpreter = ActionInterpreter(CONFIG_PATH)

    data = json.loads(Path(INPUT_JSON).read_text(encoding="utf-8"))

    for i, item in enumerate(data[:10]):  # stampa i primi 10
        obs = serializer.serialize(item)          # (1,3,10)
        action, q_vals = agent.predict(obs)       # action int, q_values (1,3)
        print(f"--- 🧪 Sample {i} ---")
        print(interpreter.humanify(action, q_values=q_vals))
        print()

if __name__ == "__main__":
    main()