Views
No views yet
1import gymnasium as gym
2
3from stable_baselines3 import PPO
4from stable_baselines3.common.env_util import make_vec_env
5from stable_baselines3.common.evaluation import evaluate_policy
6from stable_baselines3.common.monitor import Monitor
7
8env = make_vec_env('LunarLander-v2', n_envs=16)
9
10model = PPO(
11 policy='MlpPolicy',
12 env=env,
13 n_steps=1024,
14 batch_size=64,
15 n_epochs=4,
16 gamma=0.999,
17 gae_lambda=0.98,
18 ent_coef=0.01,
19 verbose=1)
20
21model.learn(total_timesteps=1000000)
22model_name = "ppo-LunarLander-v2"
23model.save(model_name)
24
25
26eval_env = Monitor(gym.make("LunarLander-v2"))
27mean_reward, std_reward = evaluate_policy(model, eval_env, n_eval_episodes=10, deterministic=True)
28print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")