Building Edge-Influenced GNN For Graph Classification

This research project contains edge-influenced graph neural network layer, GNN models.

Also training pipeline, inference pipeline, etc are added to reproduce the results.

Installation

You need to install pytorch and pytorch geometric and other supportive libraries first!. Use requirements.txt file to install the dependencies!

pip install -r requirements.txt

Usage

Adjust hyper-parameters as you see fit in the run.py

# Run the script
python3 run.py

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

• Contact: [email protected]

License

MIT