Deep reinforcement learning project 2

for course ME5406 deep learning for robotics

Table of contents

• General info
• Packages

General info

• arm_gym.py: defines the environment
• arm_unittests.py: unit tests for environment
• NN.py: DDPG implementation
• requirements.txt: required packages
• Train_test.ipynb: train and test the model

To train a new model: change the parameters in the Train_test.ipynb

To test the existing model: run the last two cells of Train_test.ipynb

Packages

Project uses Python:

• Python version: 3.6.12
• Numpy version: 1.19.1
• Matplotlib version: 3.3.1
• OpenAI gym version: 0.17.3
• Tensorflow=2.0.0
• Tensorboard=2.3.0