In this repository, we have referenced the CongestionNet paper (🔗 Kirby et al.) as well as the Graph Attention Network (GAT) paper (🔗 Veličković et al.)
Provide instructions on how to install and set up the project. Include any prerequisites or dependencies.
# Clone the repository
git clone https://github.com/lisasunhwang/Capstone-B07-3.git
# Navigate to the project directory
cd Capstone-B07-3
# Create and activate Conda environment
conda env create -f environment.yml
conda activate capstone
#preprocess data and train model
python run.py
Environment Name: capstone
Environment File: environment.yml
- matplotlib 3.8.2
- networkx 3.2.1
- numpy 1.26.4
- pandas 2.2.0
- scikit-learn 1.4.0
- scipy 1.12.0
- seaborn 0.13.2
- torch 2.2.0