Skip to content

LukeKulm/Reinforcement-Learning-Visualization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RL Algorithm Visualization

An interactive web application for visualizing and comparing different reinforcement learning algorithms in a grid world environment.

image

Features

  • Interactive grid world environment with:
    • Goal state (green)
    • Penalty state (red)
    • Agent (blue)
  • Multiple RL algorithm implementations:
    • Q-Learning
    • DQN (Deep Q-Network)
    • REINFORCE (Policy Gradient)
  • Real-time visualization of learning process
  • Cumulative reward tracking
  • Customizable hyperparameters
  • Policy visualization and playback

Installation Instructions

  1. Clone the repository:
git clone https://github.com/LukeKulm/Reinforcement-Learning-Visualization.git
cd rl-visualization
  1. Create and activate a virtual environment:

On Windows:

python -m venv venv
venv\Scripts\activate

On macOS/Linux:

python -m venv venv
source venv/bin/activate
  1. Install the required packages:
pip install -r requirements.txt

Running the Application

  1. Make sure your virtual environment is activated
  2. Start the Flask server:
python app.py
  1. Open your web browser and navigate to:
http://localhost:5000

Project Structure

rl-visualization/
├── backend/
│   ├── agents/         # RL algorithm implementations
│   ├── environment/    # Grid world environment
│   └── utils/          # Helper functions
├── frontend/
│   ├── static/         # CSS, JavaScript, and assets
│   └── templates/      # HTML templates
└── api/                # Flask API endpoints

License

This project is licensed under the MIT License - see the LICENSE file for details.

About

A visualization of how a few reinforcement learning algorithms work on a gridworld task.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors