/transit-signal-priority-with-spatial-info

Deep Reinforcement Learning with CNN networks (policy/target) training with grid (spatial) public transportation data.

Primary LanguagePythonMIT LicenseMIT

Description

Table of Contents

Environment Setup

  • Python 3.* is installed
  • Install dependencies
pip install -r requirements.txt

Training

  • After installing all the requirements, run the following command for trainig
python train.py -env cartpole
  • -env: (REQUIRED) RL Environment, availables are [cartpole, aimsun]. aimsun requires simulation software and data collecting scripts