Downloads · 30 days
1
1% of all-time downloads
smileynet/ppo-LunarLander-v2
ppo-LunarLander-v2 is a reinforcement learning model from smileynet. 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
1
1% of all-time downloads
All-time downloads
72
Public
Repo size
337 KB
Likes
0
Public
Click a slice to open those files.
.zip154 KB · 32%
From the Hugging Face model README
This is a trained model of a PPO agent playing LunarLander-v2 using the Stable-Baselines3 library.
To use this model with Stable-Baselines3, follow these steps:
import gymnasium as gym
from stable_baselines3 import PPO
# Create the environment
env = gym.make("LunarLander-v2")
# Load the trained model
model = PPO.load("path/to/model.zip")
# Run the model
obs, info = env.reset()
while True:
action, _states = model.predict(obs, deterministic=True)
obs, reward, terminated, truncated, info = env.step(action)
if terminated or truncated:
obs, info = env.reset()
env.close()
The LunarLander-v2 environment is part of the Gymnasium library.
The model was trained using the following hyperparameters:
{
'learning_rate': 0.0003,
'n_steps': 2048,
'batch_size': 64,
'n_epochs': 10,
'gamma': 0.99,
'gae_lambda': 0.95,
'clip_range': 0.2,
'ent_coef': 0.0,
'vf_coef': 0.5,
'max_grad_norm': 0.5,
}
The trained agent achieved a mean reward of 269.36 +/- 28.12 over 10 evaluation episodes.