This repository contains the official implementation of Centrality-based and Similarity-based Neighborhood Extension in Graph Neural Networks.
Parts of the code in this repository have been borrowed from the Official DGL Examples and Modules PyTorch section and Link Prediction Tutorial. It is recommended to use the test.py file to ensure the correctness of evnironment setup. The requirements for running the code are the usual machine learning setting for graphs, inlcluding: pandas, numpy, matplotlib, networkx, PyTorch, DGL. To reproduce the experiments, use the run_.py files.
This work was conducted as a joint effort with Mohammadjavad Zohrabi and Saeed Saravani under the supervision of Dr. Mostafa Chehreghani at Amirkabir University of Technology.
If you have any questions, encounter bugs, or have suggestions for improvements, please don't hesitate to contact us at [email protected] or [email protected].