/Banana_Navigation

My solution code for the first project. An agent will be trained to navigate (and collect bananas!) in a large, square world.

Primary LanguagePython

Project 1: Navigation

Introduction

For this project, you will train an agent to navigate (and collect bananas!) in a large, square world.

Random agent Trained agent
Random Agent Trained Agent

A reward of +1 is provided for collecting a yellow banana, and a reward of -1 is provided for collecting a blue banana. Thus, the goal of your agent is to collect as many yellow bananas as possible while avoiding blue bananas.

The state space has 37 dimensions and contains the agent's velocity, along with ray-based perception of objects around agent's forward direction. Given this information, the agent has to learn how to best select actions. Four discrete actions are available, corresponding to:

  • 0 - move forward.
  • 1 - move backward.
  • 2 - turn left.
  • 3 - turn right.

The task is episodic, and in order to solve the environment, your agent must get an average score of +13 over 100 consecutive episodes.

Getting Started

  1. Download the environment from one of the links below. You need only select the environment that matches your operating system:

    (For Windows users) Check out this link if you need help with determining if your computer is running a 32-bit version or 64-bit version of the Windows operating system.

    (For AWS) If you'd like to train the agent on AWS (and have not enabled a virtual screen), then please use this link to obtain the environment.

  2. Place the file in this folder, unzip (or decompress) the file and then write the correct path in the argument for creating the environment under the notebook Navigation_solution.ipynb:

env = env = UnityEnvironment(file_name="Banana.app")

Description

  • dqn_agent.py: code for the agent used in the environment
  • prioritised_double_dqn_agent.py: code for the agent, which uses prioritised double DQN
  • model.py: code containing the Q-Network used as the function approximator by the agent
  • dqn.pth: saved model weights for the original DQN model
  • ddqn.pth: saved model weights for the Double DQN model
  • dueling_dqn.pth: saved model weights for the Dueling Double DQN model
  • Navigation.ipynb: explore the unity environment & provide the solution

Instructions

Follow the instructions in Navigation.ipynb to get started with training your own agent! To watch a trained smart agent, follow the instructions below:

  • DQN: If you want to run the original DQN algorithm, use the checkpoint dqn.pth for loading the trained model. Also, choose the parameter qnetwork as QNetwork while defining the agent and the parameter update_type as dqn.
  • Double DQN: If you want to run the Double DQN algorithm, use the checkpoint double_dqn.pth for loading the trained model. Also, choose the parameter qnetwork as QNetwork while defining the agent and the parameter update_type as double_dqn.
  • Dueling Double DQN: If you want to run the Dueling Double DQN algorithm, use the checkpoint dueling_dqn.pth for loading the trained model. Also, choose the parameter qnetwork as DuelingQNetwork while defining the agent and the parameter update_type as double_dqn.
  • Prioritised Double DQN: If you want to run the Prioritised Double DQN algorithm, use the checkpoint prioritised_dqn.pth for loading the trained model.

Enhancements

Several enhancements to the original DQN algorithm have also been incorporated:

Results

Plot showing the score per episode over all the episodes. The environment was solved in 377 episodes (currently).

Double DQN DQN Dueling DQN
double-dqn-scores dqn-scores dueling-double-dqn-scores

Dependencies

Use the requirements.txt file to install the required dependencies via pip.

pip install -r requirements.txt