Contrib package of Stable Baselines3, experimental code.
Project description
Stable-Baselines3 - Contrib (SB3-Contrib)
Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code. "sb3-contrib" for short.
What is SB3-Contrib?
A place for RL algorithms and tools that are considered experimental, e.g. implementations of the latest publications. Goal is to keep the simplicity, documentation and style of stable-baselines3 but for less matured implementations.
Why create this repository?
Over the span of stable-baselines and stable-baselines3, the community has been eager to contribute in form of better logging utilities, environment wrappers, extended support (e.g. different action spaces) and learning algorithms.
However sometimes these utilities were too niche to be considered for stable-baselines or proved to be too difficult to integrate well into the existing code without creating a mess. sb3-contrib aims to fix this by not requiring the neatest code integration with existing code and not setting limits on what is too niche: almost everything remotely useful goes! We hope this allows us to provide reliable implementations following stable-baselines usual standards (consistent style, documentation, etc) beyond the relatively small scope of utilities in the main repository.
Features
See documentation for the full list of included features.
RL Algorithms:
- Truncated Quantile Critics (TQC)
- Quantile Regression DQN (QR-DQN)
- PPO with invalid action masking (MaskablePPO)
- Trust Region Policy Optimization (TRPO)
- Augmented Random Search (ARS)
Gym Wrappers:
Documentation
Documentation is available online: https://sb3-contrib.readthedocs.io/
Installation
Note: You need the master
version of Stable Baselines3.
To install Stable Baselines3 master
version:
pip install git+https://github.com/DLR-RM/stable-baselines3
To install Stable Baselines3 contrib master
version:
pip install git+https://github.com/Stable-Baselines-Team/stable-baselines3-contrib
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for sb3_contrib-2.3.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 79dadda884f2242bbab11fdb06ccf00eeead2937f7ed09ba577f803f2e52c8e1 |
|
MD5 | a27422fe2933259a75d8896fe89c0611 |
|
BLAKE2b-256 | 406058163d23588509c8c5415f5ad15164dee08e390f2e96802b8d8828f3904e |