Skip to content

qchaldemer/navigation-agent

Repository files navigation

Project 1: Navigation

Introduction

In this project, I have trained an agent to navigate (and collect bananas!) in a large, square world.

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 the 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. The environment is solved when the agent achieve an average of 13 or above on 100 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 GitHub repository

Instructions

Follow the instructions in Navigation.ipynb to see how the agent was trained and visualize it acting in the environment

Files

  • Navigation.ipynb: notebook to train and visualize the agent
  • dqn_agent.py: contains the Deep Q Network agent
  • model.py: contains the neural network to approximate the action value function
  • checkpoint.pth: contains the weight of the trained agent

About

Use Deep Q Network to train a navigation agent

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published