keras-rl  by keras-rl

Deep reinforcement learning algorithms in Python, Keras-integrated

Created 9 years ago
5,556 stars

Top 9.2% on SourcePulse

GitHubView on GitHub
Project Summary

Keras-RL provides a collection of state-of-the-art deep reinforcement learning algorithms implemented in Python, designed for seamless integration with the Keras deep learning library. It targets researchers and developers looking to experiment with and extend RL algorithms, offering easy integration with OpenAI Gym environments for rapid prototyping and evaluation.

How It Works

The library implements various RL algorithms by extending abstract classes, allowing users to easily define custom environments and agents. It leverages Keras for neural network construction and training, enabling the use of built-in or custom Keras callbacks and metrics for monitoring and controlling the learning process.

Quick Start & Requirements

  • Install via pip: pip install keras-rl
  • Additional dependencies for examples: gym, h5py, Pillow, gym[atari]
  • Example usage: python examples/dqn_cartpole.py
  • Documentation: keras-rl documentation

Highlighted Details

  • Implements DQN, Double DQN, DDPG, Continuous DQN (NAF), CEM, Dueling DQN, and Deep SARSA.
  • Supports integration with Weights & Biases (wandb) for training visualization via WandbLogger callback.
  • Allows easy extension for custom environments and algorithms.
  • Sample weights are available for download.

Maintenance & Community

  • The project is hosted on GitHub.
  • Users can file issues or submit pull requests for contributions.

Licensing & Compatibility

  • The library is released under the MIT License.
  • Compatible with Keras and OpenAI Gym.

Limitations & Caveats

  • Asynchronous Advantage Actor-Critic (A3C) and Proximal Policy Optimization (PPO) algorithms are listed as not yet implemented.
  • The project has not been updated since 2017, indicating potential compatibility issues with newer Keras or TensorFlow versions.
Health Check
Last Commit

2 years ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
0
Star History
3 stars in the last 30 days

Explore Similar Projects

Starred by Deepak Pathak Deepak Pathak(Cofounder of Skild AI; Professor at CMU), Anastasis Germanidis Anastasis Germanidis(Cofounder of Runway), and
1 more.

deer by VinF

0%
489
Deep reinforcement learning framework
Created 9 years ago
Updated 2 months ago
Feedback? Help us improve.