This is the GitHub repository for our CS4350 project, "Hub Laplacian Operators for Directional GNNs."
The complete project report, Report_GNN_project.pdf, can be found here.
This repository contains implementations of two models: our custom Graph Convolutional Neural Network (GCNN) and a modified version of Graph Attention Diffusion (GAD).
To run the implemented GCNN:
- Navigate to the
GCNN/directory. - Open
main.pyand modify the script to choose your desired hyperparameters. - Execute the script.
To run the modified GAD model:
- Go to
GAD/experiments/QM9/. - Refer to the
README.mdfile within this directory for detailed instructions and ready-to-use sample commands.
To run the spectral analysis of the dataset (or a subset of it) and obtain the cosine similarity between the different F matrices corresponding to each Hub Operator:
- Go to
GAD/experiments/QM9/. - Modify the hyperparameters in
dataset/evaluation.pyand run.
Due to space constraints in the main report, we performed an additional ablation study whose results could not be included. These results can be found in the following file:
GAD/experiments/QM9/saved_models/prop0_ablation_study/results.txt