cavis/rl4j
agibsonccc 665859d365 Update pom.xml 2021-03-19 21:23:38 +09:00
..
docs/images Fix formatting, remove obsolete files (#439) 2020-05-29 11:01:02 +03:00
rl4j-ale Unify nd4j test profiles, get rid of old modules, fix more parameter issues with junit 5 tests 2021-03-18 10:58:50 +09:00
rl4j-api Unify nd4j test profiles, get rid of old modules, fix more parameter issues with junit 5 tests 2021-03-18 10:58:50 +09:00
rl4j-core Update pom.xml 2021-03-19 21:23:38 +09:00
rl4j-doom Unify nd4j test profiles, get rid of old modules, fix more parameter issues with junit 5 tests 2021-03-18 10:58:50 +09:00
rl4j-gym Comment out more file related tests 2021-03-19 19:17:32 +09:00
rl4j-malmo Unify nd4j test profiles, get rid of old modules, fix more parameter issues with junit 5 tests 2021-03-18 10:58:50 +09:00
README.md Fix formatting, remove obsolete files (#439) 2020-05-29 11:01:02 +03:00
pom.xml Unify nd4j test profiles, get rid of old modules, fix more parameter issues with junit 5 tests 2021-03-18 10:58:50 +09:00

README.md

RL4J: Reinforcement Learning for Java

RL4J is a reinforcement learning framework integrated with deeplearning4j and released under an Apache 2.0 open-source license. By contributing code to this repository, you agree to make your contribution available under an Apache 2.0 license.

  • DQN (Deep Q Learning with double DQN)
  • Async RL (A3C, Async NStepQlearning)

Both for Low-Dimensional (array of info) and high-dimensional (pixels) input.

DOOM

Cartpole

Here is a useful blog post I wrote to introduce you to reinforcement learning, DQN and Async RL:

Blog post

Examples

Cartpole example

Disclaimer

This is a tech preview and distributed as is. Comments are welcome on our gitter channel: gitter

Quickstart

  • mvn install

Visualisation

webapp-rl4j

Quicktry cartpole:

  • run with this main

Doom

Doom is not ready yet but you can make it work if you feel adventurous with some additional steps:

  • You will need vizdoom, compile the native lib and move it into the root of your project in a folder
  • export MAVEN_OPTS=-Djava.library.path=THEFOLDEROFTHELIB
  • mvn compile exec:java -Dexec.mainClass="YOURMAINCLASS"

Malmo (Minecraft)

Malmo

  • Download and unzip Malmo from here
  • export MALMO_HOME=YOURMALMO_FOLDER
  • export MALMO_XSD_PATH=$MALMO_HOME/Schemas
  • launch malmo per instructions
  • run with this main

WIP

  • Documentation
  • Serialization/Deserialization (load save)
  • Compression of pixels in order to store 1M state in a reasonnable amount of memory
  • Async learning: A3C and nstep learning (requires some missing features from dl4j (calc and apply gradients)).

Author

Ruben Fiszel

Proposed contribution area:

  • Continuous control
  • Policy Gradient
  • Update rl4j-gym to make it compatible with pixels environments to play with Pong, Doom, etc ...