├── Awesome-Auto-GNNs.md ├── Awesome-GNN-Tricks.md ├── Awesome-Scalable-GNNs.md ├── Awsome-GNN-Acceleration.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── examples ├── gamlp_products.py ├── nafs_link_predict.py ├── nafs_node_cluster.py ├── sgc_pubmed.py ├── test_nas.py ├── test_nas_dist.py └── test_nodeclass_dist.py ├── img ├── AGDN_formula.jpg ├── AutoNE.jpg ├── FLAG_image.jpg ├── GraphSAGE.jpg ├── LBCE_loss.png ├── Label_Reuse.jpg ├── NDLS.jpg ├── PaSca.jpg ├── SGAP.png └── Sampling.jpg ├── pyproject.toml ├── requirements.txt ├── setup.py └── sgl ├── __init__.py ├── data ├── __init__.py ├── base_data.py ├── base_dataset.py ├── transforms.py └── utils.py ├── dataset ├── __init__.py ├── acm.py ├── actor.py ├── airports.py ├── amazon.py ├── amazon_product.py ├── aminer.py ├── choose_edge_type.py ├── coauthor.py ├── custom_dataset.py ├── dblp.py ├── dblp_original.py ├── facebook.py ├── flickr.py ├── github.py ├── imdb.py ├── karateclub.py ├── linkx_dataset.py ├── nell.py ├── ogbn.py ├── ogbn_mag.py ├── planetoid.py ├── reddit.py ├── twitch.py ├── utils.py ├── webkb.py └── wikics.py ├── etc ├── __init__.py ├── auto_select_edge_type_for_nars.py ├── hetero_search.py ├── hetero_test.py └── stability_of_subgraph_weight.py ├── models ├── __init__.py ├── base_model.py ├── base_model_dist.py ├── hetero │ ├── __init__.py │ ├── fast_nars_sgc.py │ └── nars_sign.py ├── homo │ ├── __init__.py │ ├── gamlp.py │ ├── gamlp_dist.py │ ├── gamlp_recursive.py │ ├── gbp.py │ ├── nafs.py │ ├── pasca_v1.py │ ├── pasca_v2.py │ ├── pasca_v3.py │ ├── sgc.py │ ├── sgc_dist.py │ ├── sign.py │ └── ssgc.py └── simple_models.py ├── operators ├── __init__.py ├── base_op.py ├── csrc │ ├── cudamatmul.c │ ├── cudamatmul.h │ ├── cudamatmul.o │ ├── libcudamatmul.so │ ├── libmatmul.so │ ├── matmul.c │ └── matmul.h ├── graph_op │ ├── __init__.py │ ├── laplacian_graph_op.py │ └── ppr_graph_op.py ├── message_op │ ├── __init__.py │ ├── concat_message_op.py │ ├── iterate_learnable_weighted_message_op.py │ ├── last_message_op.py │ ├── learnable_weighted_messahe_op.py │ ├── max_message_op.py │ ├── mean_message_op.py │ ├── min_message_op.py │ ├── over_smooth_distance_op.py │ ├── projected_concat_message_op.py │ ├── simple_weighted_message_op.py │ └── sum_message_op.py └── utils.py ├── search ├── __init__.py ├── auto_search.py ├── auto_search_dist.py ├── base_search.py ├── search_config.py ├── search_config_dist.py ├── search_models.py ├── search_models_dist.py └── utils.py ├── tasks ├── __init__.py ├── base_task.py ├── clustering_metrics.py ├── correct_and_smooth.py ├── link_prediction.py ├── node_classification.py ├── node_classification_dist.py ├── node_classification_with_label_use.py ├── node_clustering.py └── utils.py ├── tricks ├── __init__.py ├── correct_and_smooth.py └── utils.py └── utils ├── __init__.py └── auto_choose_gpu.py /Awesome-Auto-GNNs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/Awesome-Auto-GNNs.md -------------------------------------------------------------------------------- /Awesome-GNN-Tricks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/Awesome-GNN-Tricks.md -------------------------------------------------------------------------------- /Awesome-Scalable-GNNs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/Awesome-Scalable-GNNs.md -------------------------------------------------------------------------------- /Awsome-GNN-Acceleration.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/Awsome-GNN-Acceleration.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/README.md -------------------------------------------------------------------------------- /examples/gamlp_products.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/gamlp_products.py -------------------------------------------------------------------------------- /examples/nafs_link_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/nafs_link_predict.py -------------------------------------------------------------------------------- /examples/nafs_node_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/nafs_node_cluster.py -------------------------------------------------------------------------------- /examples/sgc_pubmed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/sgc_pubmed.py -------------------------------------------------------------------------------- /examples/test_nas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/test_nas.py -------------------------------------------------------------------------------- /examples/test_nas_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/test_nas_dist.py -------------------------------------------------------------------------------- /examples/test_nodeclass_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/examples/test_nodeclass_dist.py -------------------------------------------------------------------------------- /img/AGDN_formula.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/AGDN_formula.jpg -------------------------------------------------------------------------------- /img/AutoNE.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/AutoNE.jpg -------------------------------------------------------------------------------- /img/FLAG_image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/FLAG_image.jpg -------------------------------------------------------------------------------- /img/GraphSAGE.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/GraphSAGE.jpg -------------------------------------------------------------------------------- /img/LBCE_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/LBCE_loss.png -------------------------------------------------------------------------------- /img/Label_Reuse.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/Label_Reuse.jpg -------------------------------------------------------------------------------- /img/NDLS.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/NDLS.jpg -------------------------------------------------------------------------------- /img/PaSca.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/PaSca.jpg -------------------------------------------------------------------------------- /img/SGAP.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/SGAP.png -------------------------------------------------------------------------------- /img/Sampling.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/img/Sampling.jpg -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/setup.py -------------------------------------------------------------------------------- /sgl/__init__.py: -------------------------------------------------------------------------------- 1 | name = "sgl" 2 | -------------------------------------------------------------------------------- /sgl/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/data/__init__.py -------------------------------------------------------------------------------- /sgl/data/base_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/data/base_data.py -------------------------------------------------------------------------------- /sgl/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/data/base_dataset.py -------------------------------------------------------------------------------- /sgl/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/data/transforms.py -------------------------------------------------------------------------------- /sgl/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/data/utils.py -------------------------------------------------------------------------------- /sgl/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/__init__.py -------------------------------------------------------------------------------- /sgl/dataset/acm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/acm.py -------------------------------------------------------------------------------- /sgl/dataset/actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/actor.py -------------------------------------------------------------------------------- /sgl/dataset/airports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/airports.py -------------------------------------------------------------------------------- /sgl/dataset/amazon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/amazon.py -------------------------------------------------------------------------------- /sgl/dataset/amazon_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/amazon_product.py -------------------------------------------------------------------------------- /sgl/dataset/aminer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/aminer.py -------------------------------------------------------------------------------- /sgl/dataset/choose_edge_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/choose_edge_type.py -------------------------------------------------------------------------------- /sgl/dataset/coauthor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/coauthor.py -------------------------------------------------------------------------------- /sgl/dataset/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/custom_dataset.py -------------------------------------------------------------------------------- /sgl/dataset/dblp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/dblp.py -------------------------------------------------------------------------------- /sgl/dataset/dblp_original.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/dblp_original.py -------------------------------------------------------------------------------- /sgl/dataset/facebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/facebook.py -------------------------------------------------------------------------------- /sgl/dataset/flickr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/flickr.py -------------------------------------------------------------------------------- /sgl/dataset/github.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/github.py -------------------------------------------------------------------------------- /sgl/dataset/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/imdb.py -------------------------------------------------------------------------------- /sgl/dataset/karateclub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/karateclub.py -------------------------------------------------------------------------------- /sgl/dataset/linkx_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/linkx_dataset.py -------------------------------------------------------------------------------- /sgl/dataset/nell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/nell.py -------------------------------------------------------------------------------- /sgl/dataset/ogbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/ogbn.py -------------------------------------------------------------------------------- /sgl/dataset/ogbn_mag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/ogbn_mag.py -------------------------------------------------------------------------------- /sgl/dataset/planetoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/planetoid.py -------------------------------------------------------------------------------- /sgl/dataset/reddit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/reddit.py -------------------------------------------------------------------------------- /sgl/dataset/twitch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/twitch.py -------------------------------------------------------------------------------- /sgl/dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/utils.py -------------------------------------------------------------------------------- /sgl/dataset/webkb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/webkb.py -------------------------------------------------------------------------------- /sgl/dataset/wikics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/dataset/wikics.py -------------------------------------------------------------------------------- /sgl/etc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgl/etc/auto_select_edge_type_for_nars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/etc/auto_select_edge_type_for_nars.py -------------------------------------------------------------------------------- /sgl/etc/hetero_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/etc/hetero_search.py -------------------------------------------------------------------------------- /sgl/etc/hetero_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/etc/hetero_test.py -------------------------------------------------------------------------------- /sgl/etc/stability_of_subgraph_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/etc/stability_of_subgraph_weight.py -------------------------------------------------------------------------------- /sgl/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgl/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/base_model.py -------------------------------------------------------------------------------- /sgl/models/base_model_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/base_model_dist.py -------------------------------------------------------------------------------- /sgl/models/hetero/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/hetero/__init__.py -------------------------------------------------------------------------------- /sgl/models/hetero/fast_nars_sgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/hetero/fast_nars_sgc.py -------------------------------------------------------------------------------- /sgl/models/hetero/nars_sign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/hetero/nars_sign.py -------------------------------------------------------------------------------- /sgl/models/homo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/__init__.py -------------------------------------------------------------------------------- /sgl/models/homo/gamlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/gamlp.py -------------------------------------------------------------------------------- /sgl/models/homo/gamlp_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/gamlp_dist.py -------------------------------------------------------------------------------- /sgl/models/homo/gamlp_recursive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/gamlp_recursive.py -------------------------------------------------------------------------------- /sgl/models/homo/gbp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/gbp.py -------------------------------------------------------------------------------- /sgl/models/homo/nafs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/nafs.py -------------------------------------------------------------------------------- /sgl/models/homo/pasca_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/pasca_v1.py -------------------------------------------------------------------------------- /sgl/models/homo/pasca_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/pasca_v2.py -------------------------------------------------------------------------------- /sgl/models/homo/pasca_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/pasca_v3.py -------------------------------------------------------------------------------- /sgl/models/homo/sgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/sgc.py -------------------------------------------------------------------------------- /sgl/models/homo/sgc_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/sgc_dist.py -------------------------------------------------------------------------------- /sgl/models/homo/sign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/sign.py -------------------------------------------------------------------------------- /sgl/models/homo/ssgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/homo/ssgc.py -------------------------------------------------------------------------------- /sgl/models/simple_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/models/simple_models.py -------------------------------------------------------------------------------- /sgl/operators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgl/operators/base_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/base_op.py -------------------------------------------------------------------------------- /sgl/operators/csrc/cudamatmul.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/cudamatmul.c -------------------------------------------------------------------------------- /sgl/operators/csrc/cudamatmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/cudamatmul.h -------------------------------------------------------------------------------- /sgl/operators/csrc/cudamatmul.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/cudamatmul.o -------------------------------------------------------------------------------- /sgl/operators/csrc/libcudamatmul.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/libcudamatmul.so -------------------------------------------------------------------------------- /sgl/operators/csrc/libmatmul.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/libmatmul.so -------------------------------------------------------------------------------- /sgl/operators/csrc/matmul.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/matmul.c -------------------------------------------------------------------------------- /sgl/operators/csrc/matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/csrc/matmul.h -------------------------------------------------------------------------------- /sgl/operators/graph_op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/graph_op/__init__.py -------------------------------------------------------------------------------- /sgl/operators/graph_op/laplacian_graph_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/graph_op/laplacian_graph_op.py -------------------------------------------------------------------------------- /sgl/operators/graph_op/ppr_graph_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/graph_op/ppr_graph_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/__init__.py -------------------------------------------------------------------------------- /sgl/operators/message_op/concat_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/concat_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/iterate_learnable_weighted_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/iterate_learnable_weighted_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/last_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/last_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/learnable_weighted_messahe_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/learnable_weighted_messahe_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/max_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/max_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/mean_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/mean_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/min_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/min_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/over_smooth_distance_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/over_smooth_distance_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/projected_concat_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/projected_concat_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/simple_weighted_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/simple_weighted_message_op.py -------------------------------------------------------------------------------- /sgl/operators/message_op/sum_message_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/message_op/sum_message_op.py -------------------------------------------------------------------------------- /sgl/operators/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/operators/utils.py -------------------------------------------------------------------------------- /sgl/search/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sgl/search/auto_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/auto_search.py -------------------------------------------------------------------------------- /sgl/search/auto_search_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/auto_search_dist.py -------------------------------------------------------------------------------- /sgl/search/base_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/base_search.py -------------------------------------------------------------------------------- /sgl/search/search_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/search_config.py -------------------------------------------------------------------------------- /sgl/search/search_config_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/search_config_dist.py -------------------------------------------------------------------------------- /sgl/search/search_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/search_models.py -------------------------------------------------------------------------------- /sgl/search/search_models_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/search_models_dist.py -------------------------------------------------------------------------------- /sgl/search/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/search/utils.py -------------------------------------------------------------------------------- /sgl/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/__init__.py -------------------------------------------------------------------------------- /sgl/tasks/base_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/base_task.py -------------------------------------------------------------------------------- /sgl/tasks/clustering_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/clustering_metrics.py -------------------------------------------------------------------------------- /sgl/tasks/correct_and_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/correct_and_smooth.py -------------------------------------------------------------------------------- /sgl/tasks/link_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/link_prediction.py -------------------------------------------------------------------------------- /sgl/tasks/node_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/node_classification.py -------------------------------------------------------------------------------- /sgl/tasks/node_classification_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/node_classification_dist.py -------------------------------------------------------------------------------- /sgl/tasks/node_classification_with_label_use.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/node_classification_with_label_use.py -------------------------------------------------------------------------------- /sgl/tasks/node_clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/node_clustering.py -------------------------------------------------------------------------------- /sgl/tasks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tasks/utils.py -------------------------------------------------------------------------------- /sgl/tricks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tricks/__init__.py -------------------------------------------------------------------------------- /sgl/tricks/correct_and_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tricks/correct_and_smooth.py -------------------------------------------------------------------------------- /sgl/tricks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/tricks/utils.py -------------------------------------------------------------------------------- /sgl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/utils/__init__.py -------------------------------------------------------------------------------- /sgl/utils/auto_choose_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-DAIR/SGL/HEAD/sgl/utils/auto_choose_gpu.py --------------------------------------------------------------------------------