how to make this code work for a simple undirected graph ?

[binonteji]

Please help ?

[raunakkmr]

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.