Views
No views yet
1import torch
2import gymnasium as gym
3from dqn_agent import DQNAgent
4
5# 1. Initialize environment
6env = gym.make("LunarLander-v3", render_mode="human")
7state, _ = env.reset()
8
9# 2. Load Agent
10agent = DQNAgent(state_dim=8, action_dim=4)
11agent.load("best_lunar_lander_dqn.pth")
12
13# 3. Simulate
14total_reward = 0
15done = False
16while not done:
17 action, _ = agent.select_action(state, evaluate=True)
18 next_state, reward, terminated, truncated, _ = env.step(action)
19 done = terminated or truncated
20 state = next_state
21 total_reward += reward
22
23print(f"Final Landing Reward: {total_reward:.2f}")
24env.close()