Automatic code generator for training Reinforcement Learning policies

Automatic code generator for training Reinforcement Learning policies visual
rl-traingenerator demo

Generate custom template code to train your reinforcement learning policy using a simple web UI built with Streamlit. It includes different environments and can be expanded to support multiple policies and frameworks with a high level of flexible hyperparameters customization.

The generated code can be easily downloaded as .py file or Jupyter Notebook so as to immediately start training your model or use it as a baseline for your project. The backbone of this project has been taken from this repository where you can contribute with your own template.

References

Leave a comment