GraphSAGE-and-GAT-for-link-prediction
GraphSAGE-and-GAT-for-link-prediction copied to clipboard
Published
20 hours ago •
raunakkmr
raunakkmr
how to make this code work for a simple undirected graph ?
Although the models themselves do not make use of temporal information, the datasets that we use are temporal networks obtained from SNAP and Network Repository. The datasets that I tried can be found through the class names in src/datasets/link_prediction.py.
The datasets are processed in src/dataset/link_prediction.py. You can go through the code here to figure out how to process raw data from an undirected graph differently.
