mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-14 20:58:03 +00:00
45 lines
1.1 KiB
ReStructuredText
45 lines
1.1 KiB
ReStructuredText
.. _quickstart:
|
|
|
|
===============
|
|
Getting Started
|
|
===============
|
|
|
|
Most of the library tries to follow a sklearn-like syntax for the Reinforcement Learning algorithms.
|
|
|
|
Here is a quick example of how to train and run A2C on a CartPole environment:
|
|
|
|
.. code-block:: python
|
|
|
|
import gym
|
|
|
|
from stable_baselines3 import A2C
|
|
|
|
env = gym.make("CartPole-v1")
|
|
|
|
model = A2C("MlpPolicy", env, verbose=1)
|
|
model.learn(total_timesteps=10_000)
|
|
|
|
vec_env = model.get_env()
|
|
obs = vec_env.reset()
|
|
for i in range(1000):
|
|
action, _state = model.predict(obs, deterministic=True)
|
|
obs, reward, done, info = vec_env.step(action)
|
|
vec_env.render()
|
|
# VecEnv resets automatically
|
|
# if done:
|
|
# obs = vec_env.reset()
|
|
|
|
.. note::
|
|
|
|
You can find explanations about the logger output and names in the :ref:`Logger <logger>` section.
|
|
|
|
|
|
Or just train a model with a one liner if
|
|
`the environment is registered in Gym <https://github.com/openai/gym/wiki/Environments>`_ and if
|
|
the policy is registered:
|
|
|
|
.. code-block:: python
|
|
|
|
from stable_baselines3 import A2C
|
|
|
|
model = A2C("MlpPolicy", "CartPole-v1").learn(10000)
|