What problem does it solve?
This Skill solves the challenge of implementing and training various reinforcement learning (RL) agents. It offers a comprehensive toolkit to streamline RL experiments, prototyping, and model implementations, especially for single-agent RL with Gymnasium environments.
Core Features & Use Cases
- Reinforcement Learning Algorithms: Offers production-ready algorithms like PPO, SAC, DQN, TD3, DDPG, A2C, etc.
- Unified API: Facilitates easy experimentation and rapid prototyping with a scikit-learn-like interface.
- Gymnasium Compatibility: Best suited for RL tasks with Gymnasium environments, supporting single-agent RL.
- Use Case: Imagine you want to implement a custom environment for a new game. You can use this Skill to define the environment and train agents like DQN or PPO with ease.
Quick Start
Use the stable-baselines3 skill to train a reinforcement learning agent on your custom Gymnasium environment. First, create your environment by extending gymnasium.Env and define your action and observation spaces. Then, instantiate an agent using the desired algorithm, train it with the appropriate number of timesteps, and evaluate its performance.