├── .gitignore ├── LICENSE ├── README.md ├── best_parameters.json ├── data ├── README ├── edge_probabilities │ ├── cora_graph_2_logits.pkl │ └── cora_graph_5_logits.pkl └── graphs │ ├── airport_adj.pkl │ ├── airport_features.pkl │ ├── airport_labels.pkl │ ├── airport_tvt_nids.pkl │ ├── blogcatalog_adj.pkl │ ├── blogcatalog_features.pkl │ ├── blogcatalog_labels.pkl │ ├── blogcatalog_tvt_nids.pkl │ ├── citeseer_adj.pkl │ ├── citeseer_features.pkl │ ├── citeseer_labels.pkl │ ├── citeseer_tvt_nids.pkl │ ├── cora_adj.pkl │ ├── cora_features.pkl │ ├── cora_labels.pkl │ ├── cora_tvt_nids.pkl │ ├── flickr_adj.pkl │ ├── flickr_features.pkl │ ├── flickr_labels.pkl │ ├── flickr_tvt_nids.pkl │ ├── ppi_adj.pkl │ ├── ppi_features.pkl │ ├── ppi_labels.pkl │ └── ppi_tvt_nids.pkl ├── models ├── GAT_dgl.py ├── GAug.py ├── GAugO_minibatch.py ├── GCN_dgl.py ├── GSAGE_dgl.py └── JKNet_dgl.py ├── optuna_GAugM.py ├── optuna_GAugO.py ├── requirements.txt ├── train_GAugM.py ├── train_GAugO.py └── vgae ├── README.md ├── dataloader.py ├── main.py ├── models.py ├── run.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/README.md -------------------------------------------------------------------------------- /best_parameters.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/best_parameters.json -------------------------------------------------------------------------------- /data/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/README -------------------------------------------------------------------------------- /data/edge_probabilities/cora_graph_2_logits.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/edge_probabilities/cora_graph_2_logits.pkl -------------------------------------------------------------------------------- /data/edge_probabilities/cora_graph_5_logits.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/edge_probabilities/cora_graph_5_logits.pkl -------------------------------------------------------------------------------- /data/graphs/airport_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/airport_adj.pkl -------------------------------------------------------------------------------- /data/graphs/airport_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/airport_features.pkl -------------------------------------------------------------------------------- /data/graphs/airport_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/airport_labels.pkl -------------------------------------------------------------------------------- /data/graphs/airport_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/airport_tvt_nids.pkl -------------------------------------------------------------------------------- /data/graphs/blogcatalog_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/blogcatalog_adj.pkl -------------------------------------------------------------------------------- /data/graphs/blogcatalog_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/blogcatalog_features.pkl -------------------------------------------------------------------------------- /data/graphs/blogcatalog_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/blogcatalog_labels.pkl -------------------------------------------------------------------------------- /data/graphs/blogcatalog_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/blogcatalog_tvt_nids.pkl -------------------------------------------------------------------------------- /data/graphs/citeseer_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/citeseer_adj.pkl -------------------------------------------------------------------------------- /data/graphs/citeseer_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/citeseer_features.pkl -------------------------------------------------------------------------------- /data/graphs/citeseer_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/citeseer_labels.pkl -------------------------------------------------------------------------------- /data/graphs/citeseer_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/citeseer_tvt_nids.pkl -------------------------------------------------------------------------------- /data/graphs/cora_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/cora_adj.pkl -------------------------------------------------------------------------------- /data/graphs/cora_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/cora_features.pkl -------------------------------------------------------------------------------- /data/graphs/cora_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/cora_labels.pkl -------------------------------------------------------------------------------- /data/graphs/cora_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/cora_tvt_nids.pkl -------------------------------------------------------------------------------- /data/graphs/flickr_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/flickr_adj.pkl -------------------------------------------------------------------------------- /data/graphs/flickr_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/flickr_features.pkl -------------------------------------------------------------------------------- /data/graphs/flickr_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/flickr_labels.pkl -------------------------------------------------------------------------------- /data/graphs/flickr_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/flickr_tvt_nids.pkl -------------------------------------------------------------------------------- /data/graphs/ppi_adj.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/ppi_adj.pkl -------------------------------------------------------------------------------- /data/graphs/ppi_features.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/ppi_features.pkl -------------------------------------------------------------------------------- /data/graphs/ppi_labels.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/ppi_labels.pkl -------------------------------------------------------------------------------- /data/graphs/ppi_tvt_nids.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/data/graphs/ppi_tvt_nids.pkl -------------------------------------------------------------------------------- /models/GAT_dgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/GAT_dgl.py -------------------------------------------------------------------------------- /models/GAug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/GAug.py -------------------------------------------------------------------------------- /models/GAugO_minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/GAugO_minibatch.py -------------------------------------------------------------------------------- /models/GCN_dgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/GCN_dgl.py -------------------------------------------------------------------------------- /models/GSAGE_dgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/GSAGE_dgl.py -------------------------------------------------------------------------------- /models/JKNet_dgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/models/JKNet_dgl.py -------------------------------------------------------------------------------- /optuna_GAugM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/optuna_GAugM.py -------------------------------------------------------------------------------- /optuna_GAugO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/optuna_GAugO.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_GAugM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/train_GAugM.py -------------------------------------------------------------------------------- /train_GAugO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/train_GAugO.py -------------------------------------------------------------------------------- /vgae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/README.md -------------------------------------------------------------------------------- /vgae/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/dataloader.py -------------------------------------------------------------------------------- /vgae/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/main.py -------------------------------------------------------------------------------- /vgae/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/models.py -------------------------------------------------------------------------------- /vgae/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/run.sh -------------------------------------------------------------------------------- /vgae/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhao-tong/GAug/HEAD/vgae/utils.py --------------------------------------------------------------------------------