├── .github └── workflows │ └── ci.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── CONTRIBUTE.md ├── LICENSE ├── README.md ├── conda_env.yml ├── docs ├── Makefile ├── README.md └── source │ ├── 404.rst │ ├── _static │ ├── css │ │ └── custom.css │ ├── favicon.svg │ └── img │ │ ├── logos │ │ └── pyg.svg │ │ ├── sel-1overk.png │ │ ├── sel-cluster.png │ │ ├── sel-scoring.png │ │ ├── src_overview.png │ │ ├── srcl.png │ │ ├── tgp-logo-bar.svg │ │ ├── tgp-logo-ext.svg │ │ ├── tgp-logo-full.svg │ │ └── tgp-logo.svg │ ├── _templates │ ├── components │ │ └── foot-copyright.html │ └── partials │ │ └── globaltoc-above.html │ ├── api │ ├── connect.rst │ ├── data │ │ ├── index.rst │ │ ├── loaders.rst │ │ └── transforms.rst │ ├── datasets.rst │ ├── lift.rst │ ├── poolers.rst │ ├── reduce.rst │ ├── select.rst │ ├── src.rst │ ├── tgp.rst │ └── utils │ │ ├── index.rst │ │ ├── losses.rst │ │ └── typing.rst │ ├── conf.py │ ├── content │ ├── quickstart.rst │ └── src.md │ ├── genindex.rst │ ├── index.rst │ ├── poolers_cheatsheet.rst │ └── tutorials │ ├── advanced.ipynb │ ├── index.rst │ ├── intro.ipynb │ └── preprocessing_and_transforms.ipynb ├── examples ├── classification.py ├── classification_pan.py ├── clustering.py ├── clustering_tvgnn.py ├── node_class.py └── pre_coarsening.py ├── pre-requirements.txt ├── pyproject.toml ├── tests ├── .gitkeep ├── test_asap.py ├── test_base_lift.py ├── test_base_select.py ├── test_bnpool.py ├── test_bnpool_losses.py ├── test_cheatsheet.py ├── test_connect.py ├── test_diffpool.py ├── test_dmon.py ├── test_ecpool.py ├── test_edge_weight_normalization.py ├── test_global_reduce.py ├── test_graclus.py ├── test_hoscpool.py ├── test_imports.py ├── test_justbalance.py ├── test_kmis.py ├── test_lapool.py ├── test_loaders.py ├── test_maxcut.py ├── test_mincut_loss.py ├── test_mp.py ├── test_ndp.py ├── test_nmf.py ├── test_nopool.py ├── test_poolers.py ├── test_precoarsening.py ├── test_signature.py ├── test_spt_adj.py ├── test_src.py ├── test_topk.py ├── test_trainable.py └── test_transforms.py └── tgp ├── __init__.py ├── connect ├── __init__.py ├── base_conn.py ├── dense_conn.py ├── dense_conn_spt.py └── kron_conn.py ├── data ├── __init__.py ├── collate.py ├── loaders.py └── transforms.py ├── datasets ├── __init__.py ├── expwl1.py ├── graph_classification_bench.py ├── gset.py ├── multipartite_graph.py └── pygsp.py ├── imports.py ├── lift ├── __init__.py └── base_lift.py ├── mp ├── __init__.py └── gtvconv.py ├── poolers ├── __init__.py ├── asap.py ├── asym_cheeger_cut.py ├── bnpool.py ├── diffpool.py ├── dmon.py ├── edge_contraction.py ├── graclus.py ├── hosc.py ├── just_balance.py ├── kmis.py ├── lapool.py ├── maxcut.py ├── mincut.py ├── ndp.py ├── nmf.py ├── nopool.py ├── pan.py ├── sag.py └── topk.py ├── reduce ├── __init__.py ├── base_reduce.py └── global_reduce.py ├── select ├── __init__.py ├── base_select.py ├── dense_select.py ├── dp_select.py ├── edge_contraction_select.py ├── graclus_select.py ├── identity_select.py ├── kmis_select.py ├── lapool_select.py ├── maxcut_select.py ├── ndp_select.py ├── nmf_select.py └── topk_select.py ├── src.py └── utils ├── __init__.py ├── cheatsheet.py ├── losses.py ├── ops.py ├── signature.py └── typing.py /.github/workflows/ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/.github/workflows/ci.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /CONTRIBUTE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/CONTRIBUTE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/README.md -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/conda_env.yml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/source/404.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/404.rst -------------------------------------------------------------------------------- /docs/source/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/css/custom.css -------------------------------------------------------------------------------- /docs/source/_static/favicon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/favicon.svg -------------------------------------------------------------------------------- /docs/source/_static/img/logos/pyg.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/logos/pyg.svg -------------------------------------------------------------------------------- /docs/source/_static/img/sel-1overk.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/sel-1overk.png -------------------------------------------------------------------------------- /docs/source/_static/img/sel-cluster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/sel-cluster.png -------------------------------------------------------------------------------- /docs/source/_static/img/sel-scoring.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/sel-scoring.png -------------------------------------------------------------------------------- /docs/source/_static/img/src_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/src_overview.png -------------------------------------------------------------------------------- /docs/source/_static/img/srcl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/srcl.png -------------------------------------------------------------------------------- /docs/source/_static/img/tgp-logo-bar.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/tgp-logo-bar.svg -------------------------------------------------------------------------------- /docs/source/_static/img/tgp-logo-ext.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/tgp-logo-ext.svg -------------------------------------------------------------------------------- /docs/source/_static/img/tgp-logo-full.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/tgp-logo-full.svg -------------------------------------------------------------------------------- /docs/source/_static/img/tgp-logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_static/img/tgp-logo.svg -------------------------------------------------------------------------------- /docs/source/_templates/components/foot-copyright.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_templates/components/foot-copyright.html -------------------------------------------------------------------------------- /docs/source/_templates/partials/globaltoc-above.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/_templates/partials/globaltoc-above.html -------------------------------------------------------------------------------- /docs/source/api/connect.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/connect.rst -------------------------------------------------------------------------------- /docs/source/api/data/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/data/index.rst -------------------------------------------------------------------------------- /docs/source/api/data/loaders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/data/loaders.rst -------------------------------------------------------------------------------- /docs/source/api/data/transforms.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/data/transforms.rst -------------------------------------------------------------------------------- /docs/source/api/datasets.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/datasets.rst -------------------------------------------------------------------------------- /docs/source/api/lift.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/lift.rst -------------------------------------------------------------------------------- /docs/source/api/poolers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/poolers.rst -------------------------------------------------------------------------------- /docs/source/api/reduce.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/reduce.rst -------------------------------------------------------------------------------- /docs/source/api/select.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/select.rst -------------------------------------------------------------------------------- /docs/source/api/src.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/src.rst -------------------------------------------------------------------------------- /docs/source/api/tgp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/tgp.rst -------------------------------------------------------------------------------- /docs/source/api/utils/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/utils/index.rst -------------------------------------------------------------------------------- /docs/source/api/utils/losses.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/utils/losses.rst -------------------------------------------------------------------------------- /docs/source/api/utils/typing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/api/utils/typing.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/content/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/content/quickstart.rst -------------------------------------------------------------------------------- /docs/source/content/src.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/content/src.md -------------------------------------------------------------------------------- /docs/source/genindex.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/genindex.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/poolers_cheatsheet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/poolers_cheatsheet.rst -------------------------------------------------------------------------------- /docs/source/tutorials/advanced.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/tutorials/advanced.ipynb -------------------------------------------------------------------------------- /docs/source/tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/tutorials/index.rst -------------------------------------------------------------------------------- /docs/source/tutorials/intro.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/tutorials/intro.ipynb -------------------------------------------------------------------------------- /docs/source/tutorials/preprocessing_and_transforms.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/docs/source/tutorials/preprocessing_and_transforms.ipynb -------------------------------------------------------------------------------- /examples/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/classification.py -------------------------------------------------------------------------------- /examples/classification_pan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/classification_pan.py -------------------------------------------------------------------------------- /examples/clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/clustering.py -------------------------------------------------------------------------------- /examples/clustering_tvgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/clustering_tvgnn.py -------------------------------------------------------------------------------- /examples/node_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/node_class.py -------------------------------------------------------------------------------- /examples/pre_coarsening.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/examples/pre_coarsening.py -------------------------------------------------------------------------------- /pre-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/pre-requirements.txt -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_asap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_asap.py -------------------------------------------------------------------------------- /tests/test_base_lift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_base_lift.py -------------------------------------------------------------------------------- /tests/test_base_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_base_select.py -------------------------------------------------------------------------------- /tests/test_bnpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_bnpool.py -------------------------------------------------------------------------------- /tests/test_bnpool_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_bnpool_losses.py -------------------------------------------------------------------------------- /tests/test_cheatsheet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_cheatsheet.py -------------------------------------------------------------------------------- /tests/test_connect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_connect.py -------------------------------------------------------------------------------- /tests/test_diffpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_diffpool.py -------------------------------------------------------------------------------- /tests/test_dmon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_dmon.py -------------------------------------------------------------------------------- /tests/test_ecpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_ecpool.py -------------------------------------------------------------------------------- /tests/test_edge_weight_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_edge_weight_normalization.py -------------------------------------------------------------------------------- /tests/test_global_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_global_reduce.py -------------------------------------------------------------------------------- /tests/test_graclus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_graclus.py -------------------------------------------------------------------------------- /tests/test_hoscpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_hoscpool.py -------------------------------------------------------------------------------- /tests/test_imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_imports.py -------------------------------------------------------------------------------- /tests/test_justbalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_justbalance.py -------------------------------------------------------------------------------- /tests/test_kmis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_kmis.py -------------------------------------------------------------------------------- /tests/test_lapool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_lapool.py -------------------------------------------------------------------------------- /tests/test_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_loaders.py -------------------------------------------------------------------------------- /tests/test_maxcut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_maxcut.py -------------------------------------------------------------------------------- /tests/test_mincut_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_mincut_loss.py -------------------------------------------------------------------------------- /tests/test_mp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_mp.py -------------------------------------------------------------------------------- /tests/test_ndp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_ndp.py -------------------------------------------------------------------------------- /tests/test_nmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_nmf.py -------------------------------------------------------------------------------- /tests/test_nopool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_nopool.py -------------------------------------------------------------------------------- /tests/test_poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_poolers.py -------------------------------------------------------------------------------- /tests/test_precoarsening.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_precoarsening.py -------------------------------------------------------------------------------- /tests/test_signature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_signature.py -------------------------------------------------------------------------------- /tests/test_spt_adj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_spt_adj.py -------------------------------------------------------------------------------- /tests/test_src.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_src.py -------------------------------------------------------------------------------- /tests/test_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_topk.py -------------------------------------------------------------------------------- /tests/test_trainable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_trainable.py -------------------------------------------------------------------------------- /tests/test_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tests/test_transforms.py -------------------------------------------------------------------------------- /tgp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/__init__.py -------------------------------------------------------------------------------- /tgp/connect/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/connect/__init__.py -------------------------------------------------------------------------------- /tgp/connect/base_conn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/connect/base_conn.py -------------------------------------------------------------------------------- /tgp/connect/dense_conn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/connect/dense_conn.py -------------------------------------------------------------------------------- /tgp/connect/dense_conn_spt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/connect/dense_conn_spt.py -------------------------------------------------------------------------------- /tgp/connect/kron_conn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/connect/kron_conn.py -------------------------------------------------------------------------------- /tgp/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/data/__init__.py -------------------------------------------------------------------------------- /tgp/data/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/data/collate.py -------------------------------------------------------------------------------- /tgp/data/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/data/loaders.py -------------------------------------------------------------------------------- /tgp/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/data/transforms.py -------------------------------------------------------------------------------- /tgp/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/__init__.py -------------------------------------------------------------------------------- /tgp/datasets/expwl1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/expwl1.py -------------------------------------------------------------------------------- /tgp/datasets/graph_classification_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/graph_classification_bench.py -------------------------------------------------------------------------------- /tgp/datasets/gset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/gset.py -------------------------------------------------------------------------------- /tgp/datasets/multipartite_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/multipartite_graph.py -------------------------------------------------------------------------------- /tgp/datasets/pygsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/datasets/pygsp.py -------------------------------------------------------------------------------- /tgp/imports.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/imports.py -------------------------------------------------------------------------------- /tgp/lift/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/lift/__init__.py -------------------------------------------------------------------------------- /tgp/lift/base_lift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/lift/base_lift.py -------------------------------------------------------------------------------- /tgp/mp/__init__.py: -------------------------------------------------------------------------------- 1 | from .gtvconv import GTVConv 2 | -------------------------------------------------------------------------------- /tgp/mp/gtvconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/mp/gtvconv.py -------------------------------------------------------------------------------- /tgp/poolers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/__init__.py -------------------------------------------------------------------------------- /tgp/poolers/asap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/asap.py -------------------------------------------------------------------------------- /tgp/poolers/asym_cheeger_cut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/asym_cheeger_cut.py -------------------------------------------------------------------------------- /tgp/poolers/bnpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/bnpool.py -------------------------------------------------------------------------------- /tgp/poolers/diffpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/diffpool.py -------------------------------------------------------------------------------- /tgp/poolers/dmon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/dmon.py -------------------------------------------------------------------------------- /tgp/poolers/edge_contraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/edge_contraction.py -------------------------------------------------------------------------------- /tgp/poolers/graclus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/graclus.py -------------------------------------------------------------------------------- /tgp/poolers/hosc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/hosc.py -------------------------------------------------------------------------------- /tgp/poolers/just_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/just_balance.py -------------------------------------------------------------------------------- /tgp/poolers/kmis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/kmis.py -------------------------------------------------------------------------------- /tgp/poolers/lapool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/lapool.py -------------------------------------------------------------------------------- /tgp/poolers/maxcut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/maxcut.py -------------------------------------------------------------------------------- /tgp/poolers/mincut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/mincut.py -------------------------------------------------------------------------------- /tgp/poolers/ndp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/ndp.py -------------------------------------------------------------------------------- /tgp/poolers/nmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/nmf.py -------------------------------------------------------------------------------- /tgp/poolers/nopool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/nopool.py -------------------------------------------------------------------------------- /tgp/poolers/pan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/pan.py -------------------------------------------------------------------------------- /tgp/poolers/sag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/sag.py -------------------------------------------------------------------------------- /tgp/poolers/topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/poolers/topk.py -------------------------------------------------------------------------------- /tgp/reduce/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/reduce/__init__.py -------------------------------------------------------------------------------- /tgp/reduce/base_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/reduce/base_reduce.py -------------------------------------------------------------------------------- /tgp/reduce/global_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/reduce/global_reduce.py -------------------------------------------------------------------------------- /tgp/select/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/__init__.py -------------------------------------------------------------------------------- /tgp/select/base_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/base_select.py -------------------------------------------------------------------------------- /tgp/select/dense_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/dense_select.py -------------------------------------------------------------------------------- /tgp/select/dp_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/dp_select.py -------------------------------------------------------------------------------- /tgp/select/edge_contraction_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/edge_contraction_select.py -------------------------------------------------------------------------------- /tgp/select/graclus_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/graclus_select.py -------------------------------------------------------------------------------- /tgp/select/identity_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/identity_select.py -------------------------------------------------------------------------------- /tgp/select/kmis_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/kmis_select.py -------------------------------------------------------------------------------- /tgp/select/lapool_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/lapool_select.py -------------------------------------------------------------------------------- /tgp/select/maxcut_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/maxcut_select.py -------------------------------------------------------------------------------- /tgp/select/ndp_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/ndp_select.py -------------------------------------------------------------------------------- /tgp/select/nmf_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/nmf_select.py -------------------------------------------------------------------------------- /tgp/select/topk_select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/select/topk_select.py -------------------------------------------------------------------------------- /tgp/src.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/src.py -------------------------------------------------------------------------------- /tgp/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/__init__.py -------------------------------------------------------------------------------- /tgp/utils/cheatsheet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/cheatsheet.py -------------------------------------------------------------------------------- /tgp/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/losses.py -------------------------------------------------------------------------------- /tgp/utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/ops.py -------------------------------------------------------------------------------- /tgp/utils/signature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/signature.py -------------------------------------------------------------------------------- /tgp/utils/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgp-team/torch-geometric-pool/HEAD/tgp/utils/typing.py --------------------------------------------------------------------------------