- pytorch==1.11.0
- pytorch-geometric
-
Run clean_data.py This script would run the preprocessing for raw data and dump a preprocessed file.
-
Run create_edge.py This script would create a fully-connected graph
-
Run train.py This script is used for training model