├── .github └── workflows │ └── python-package.yml ├── .gitignore ├── LICENSE ├── README.md ├── environment.yml ├── notebooks ├── ray_analysis.ipynb └── visualise_attention.ipynb ├── pyG_install.sh ├── requirements.txt ├── src ├── CGNN.py ├── DIGL_data.py ├── DIGL_seeds.py ├── GNN.py ├── GNN_KNN.py ├── GNN_KNN_early.py ├── GNN_early.py ├── GNN_image.py ├── base_classes.py ├── best_params.py ├── block_constant.py ├── block_constant_rewiring.py ├── block_mixed.py ├── block_transformer_attention.py ├── block_transformer_hard_attention.py ├── block_transformer_rewiring.py ├── data.py ├── data_image.py ├── deepwalk_embeddings.py ├── deepwalk_gen.sh ├── deepwalk_gen_symlinks.py ├── distances_kNN.py ├── early_stop_solver.py ├── function_GAT_attention.py ├── function_laplacian_diffusion.py ├── function_transformer_attention.py ├── graph_rewiring.py ├── heterophilic.py ├── hyperbolic_distances.py ├── model_configurations.py ├── pos_enc_factorisation.py ├── post_analysis_image.py ├── ray_tune.py ├── regularized_ODE_function.py ├── run_GNN.py ├── run_best_ray.py ├── run_explicit_implicit_exp.py ├── run_image.py ├── utils.py └── visualise_attention.py └── test ├── test_ICML_gnn.py ├── test_attention.py ├── test_attention_ode_block.py ├── test_block_mixed.py ├── test_early_stop.py ├── test_function_laplacian_diffusion.py ├── test_gnn.py ├── test_params.py ├── test_transformer_attention.py └── test_utils.py /.github/workflows/python-package.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/.github/workflows/python-package.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/environment.yml -------------------------------------------------------------------------------- /notebooks/ray_analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/notebooks/ray_analysis.ipynb -------------------------------------------------------------------------------- /notebooks/visualise_attention.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/notebooks/visualise_attention.ipynb -------------------------------------------------------------------------------- /pyG_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/pyG_install.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/CGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/CGNN.py -------------------------------------------------------------------------------- /src/DIGL_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/DIGL_data.py -------------------------------------------------------------------------------- /src/DIGL_seeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/DIGL_seeds.py -------------------------------------------------------------------------------- /src/GNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/GNN.py -------------------------------------------------------------------------------- /src/GNN_KNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/GNN_KNN.py -------------------------------------------------------------------------------- /src/GNN_KNN_early.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/GNN_KNN_early.py -------------------------------------------------------------------------------- /src/GNN_early.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/GNN_early.py -------------------------------------------------------------------------------- /src/GNN_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/GNN_image.py -------------------------------------------------------------------------------- /src/base_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/base_classes.py -------------------------------------------------------------------------------- /src/best_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/best_params.py -------------------------------------------------------------------------------- /src/block_constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_constant.py -------------------------------------------------------------------------------- /src/block_constant_rewiring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_constant_rewiring.py -------------------------------------------------------------------------------- /src/block_mixed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_mixed.py -------------------------------------------------------------------------------- /src/block_transformer_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_transformer_attention.py -------------------------------------------------------------------------------- /src/block_transformer_hard_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_transformer_hard_attention.py -------------------------------------------------------------------------------- /src/block_transformer_rewiring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/block_transformer_rewiring.py -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/data.py -------------------------------------------------------------------------------- /src/data_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/data_image.py -------------------------------------------------------------------------------- /src/deepwalk_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/deepwalk_embeddings.py -------------------------------------------------------------------------------- /src/deepwalk_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/deepwalk_gen.sh -------------------------------------------------------------------------------- /src/deepwalk_gen_symlinks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/deepwalk_gen_symlinks.py -------------------------------------------------------------------------------- /src/distances_kNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/distances_kNN.py -------------------------------------------------------------------------------- /src/early_stop_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/early_stop_solver.py -------------------------------------------------------------------------------- /src/function_GAT_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/function_GAT_attention.py -------------------------------------------------------------------------------- /src/function_laplacian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/function_laplacian_diffusion.py -------------------------------------------------------------------------------- /src/function_transformer_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/function_transformer_attention.py -------------------------------------------------------------------------------- /src/graph_rewiring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/graph_rewiring.py -------------------------------------------------------------------------------- /src/heterophilic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/heterophilic.py -------------------------------------------------------------------------------- /src/hyperbolic_distances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/hyperbolic_distances.py -------------------------------------------------------------------------------- /src/model_configurations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/model_configurations.py -------------------------------------------------------------------------------- /src/pos_enc_factorisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/pos_enc_factorisation.py -------------------------------------------------------------------------------- /src/post_analysis_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/post_analysis_image.py -------------------------------------------------------------------------------- /src/ray_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/ray_tune.py -------------------------------------------------------------------------------- /src/regularized_ODE_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/regularized_ODE_function.py -------------------------------------------------------------------------------- /src/run_GNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/run_GNN.py -------------------------------------------------------------------------------- /src/run_best_ray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/run_best_ray.py -------------------------------------------------------------------------------- /src/run_explicit_implicit_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/run_explicit_implicit_exp.py -------------------------------------------------------------------------------- /src/run_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/run_image.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/visualise_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/src/visualise_attention.py -------------------------------------------------------------------------------- /test/test_ICML_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_ICML_gnn.py -------------------------------------------------------------------------------- /test/test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_attention.py -------------------------------------------------------------------------------- /test/test_attention_ode_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_attention_ode_block.py -------------------------------------------------------------------------------- /test/test_block_mixed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_block_mixed.py -------------------------------------------------------------------------------- /test/test_early_stop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_early_stop.py -------------------------------------------------------------------------------- /test/test_function_laplacian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_function_laplacian_diffusion.py -------------------------------------------------------------------------------- /test/test_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_gnn.py -------------------------------------------------------------------------------- /test/test_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_params.py -------------------------------------------------------------------------------- /test/test_transformer_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_transformer_attention.py -------------------------------------------------------------------------------- /test/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/graph-neural-pde/HEAD/test/test_utils.py --------------------------------------------------------------------------------