Downloads · 30 days
1
25% of all-time downloads
mavleo96/ppo-lunarlander-v2
ppo-lunarlander-v2 is a reinforcement learning model from mavleo96. 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
25% of all-time downloads
All-time downloads
4
Public
Repo size
430 KB
Likes
0
Public
Click a slice to open those files.
.mp4149 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.
TODO: Add your code
from stable_baselines3 import PPO
from huggingface_sb3 import load_from_hub
import gym
# Define model repo_id and filename
repo_id = "mavleo96/rl-bots" # Change this to the actual repo if different
filename = "ppo-LunarLander-v2.zip"
# Load the model from the Hugging Face Hub
model = load_from_hub(repo_id, filename, model_class=PPO)
# Create the environment
env = gym.make("LunarLander-v2")
# Run a few episodes
obs = env.reset()
for _ in range(1000):
action, _states = model.predict(obs, deterministic=True)
obs, reward, done, info = env.step(action)
env.render()
if done:
obs = env.reset()
env.close()