bernardcheng/cs5446_pogs

CS5446 Project: Team POGS

Quickstart

conda env create -f requirements_conda.yml

Repo Structure

Saved Models

All trained models are stored in the saved folder.

Notebooks for Hyperparameter Tuning & Training of Agent

Model	Notebook Name	Description
DQN	DQN-Optuna DQN-Optuna-EnvD	Optuna Tuning
DQN	DQN DQN-EnvD	Training Script
A2C	A2C-Optuna-EnvA-mini A2C-Optuna-EnvA A2C-Optuna-EnvB-mini A2C-Optuna-EnvB A2C-Optuna-EnvC-mini A2C-Optuna-EnvC A2C-Optuna-EnvD-v2 A2C-Optuna-EnvD	Optuna Tuning + Training Script
PPO	PPO-Optuna PPO-Optuna-EnvD	Optuna Tuning
PPO	PPO PPO-EnvD	Training Script
RPPO	RecurrentPPO_Optuna	Optuna Tuning
RPPO	RecurrentPPO_EnvA RecurrentPPO_EnvB RecurrentPPO_EnvC RecurrentPPO_EnvD	Training Script