Stable Baselines3 是一个基于 PyTorch 的强化学习库,提供了多种预训练的算法和工具。以下是一些入门级的教程,帮助您快速上手 Stable Baselines3。

安装 Stable Baselines3

首先,您需要安装 Stable Baselines3。您可以使用以下命令进行安装:

pip install stable-baselines3

简单示例

以下是一个简单的示例,展示了如何使用 Stable Baselines3 来训练一个 CartPole 环境。

import gym
from stable_baselines3 import PPO

env = gym.make("CartPole-v1")
model = PPO("MlpPolicy", env, verbose=1)
model.learn(total_timesteps=10000)

更多资源

如果您需要更深入的了解 Stable Baselines3,以下是一些推荐的资源:

希望这些教程能帮助您快速上手 Stable Baselines3!🚀

CartPole