├── .gitignore ├── LICENSE ├── README.md ├── notebooks ├── data_exploration.ipynb ├── dataset_testing.ipynb ├── gat.ipynb ├── graphsage.ipynb └── simple_baseline.ipynb ├── results.md ├── src ├── config.json ├── config_gat.json ├── config_graphsage.json ├── datasets │ └── link_prediction.py ├── layers.py ├── main.py ├── models.py └── utils.py └── visualizations └── .DS_Store /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/README.md -------------------------------------------------------------------------------- /notebooks/data_exploration.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/notebooks/data_exploration.ipynb -------------------------------------------------------------------------------- /notebooks/dataset_testing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/notebooks/dataset_testing.ipynb -------------------------------------------------------------------------------- /notebooks/gat.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/notebooks/gat.ipynb -------------------------------------------------------------------------------- /notebooks/graphsage.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/notebooks/graphsage.ipynb -------------------------------------------------------------------------------- /notebooks/simple_baseline.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/notebooks/simple_baseline.ipynb -------------------------------------------------------------------------------- /results.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/results.md -------------------------------------------------------------------------------- /src/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/config.json -------------------------------------------------------------------------------- /src/config_gat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/config_gat.json -------------------------------------------------------------------------------- /src/config_graphsage.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/config_graphsage.json -------------------------------------------------------------------------------- /src/datasets/link_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/datasets/link_prediction.py -------------------------------------------------------------------------------- /src/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/layers.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/main.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/models.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/src/utils.py -------------------------------------------------------------------------------- /visualizations/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raunakkmr/GraphSAGE-and-GAT-for-link-prediction/HEAD/visualizations/.DS_Store --------------------------------------------------------------------------------