Views
No views yet
1import gym
2from stable_baselines3 import PPO
3from huggingface_sb3 import load_from_hub
4
5checkpoint = load_from_hub(
6 repo_id="dmenini/ppo-LunarLander-v2",
7 filename="ppo-LunarLander-v2.zip"
8)
9
10model = PPO.load(checkpoint)
11
12env = gym.make("LunarLander-v2")
13
14# Evaluate the agent and watch it
15eval_env = gym.make("LunarLander-v2")
16mean_reward, std_reward = evaluate_policy(
17 model, eval_env, render=True, n_eval_episodes=5, deterministic=True, warn=False
18)
19print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
20