Instructions

Get data

See datasets/external_download_link_to_data.txt for a link to a Google Drive folder that contains all of our original and preprocessed data.

Run Matrix Factorization

Open models/factorization.ipynb and run entire notebook to train and test model.

Run Node2vec

Open models/node2vec_mlp_v2.ipynb and run entire notebook to train and test model.

Run GraphSAGE and Graph Attention Network experiments

Open ALL_Experiments_MessagePassing_Models.ipynb and run entire notebook to train, tune, and test the GNN models.