Downloads · 30 days
0
0% of all-time downloads
davidhornshaw/LunarLander-v2-PPO
LunarLander-v2-PPO is a reinforcement learning model from davidhornshaw. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. It is set up for stable-baselines3.
This is a trained model of a PPO agent playing LunarLander-v2 using the stable-baselines3 library.
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
6
Public
Repo size
434 KB
Likes
0
Public
Click a slice to open those files.
.mp4153 KB · 33%
From the Hugging Face model README
This is a trained model of a PPO agent playing LunarLander-v2 using the stable-baselines3 library.
from stable_baselines3 import PPO
from huggingface_sb3 import load_from_hub
# repo info
checkpoint = load_from_hub(
repo_id="davidhornshaw/LunarLander-v2-PPO",
filename="ppo-LunarLander-v2.zip"
)
# make model
checkpoint = load_from_hub(repo_id, filename)
model = PPO.load(checkpoint, custom_objects=custom_objects, print_system_info=True)
# evaluate model
eval_env = Monitor(gym.make("LunarLander-v2", render_mode='rgb_array'))
mean_reward, std_reward = evaluate_policy(model, eval_env, n_eval_episodes=100, deterministic=True)
print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")