├── .gitignore ├── AUTHORS.rst ├── CHANGELOG.rst ├── CONTRIBUTING.rst ├── LICENSE ├── README.md ├── setup.cfg ├── setup.py ├── single_layer.png ├── src └── autobahn │ ├── __init__.py │ ├── blocks.py │ ├── datasets.py │ ├── decompositions.py │ ├── experiments │ ├── __init__.py │ ├── combo_models.py │ ├── conf │ │ ├── config_ogb.yaml │ │ ├── config_zinc.yaml │ │ ├── dataset │ │ │ ├── ogbg_hiv.yaml │ │ │ ├── ogbg_molpcba.yaml │ │ │ ├── ogbg_muv.yaml │ │ │ ├── ogbg_tox.yaml │ │ │ ├── zinc_full.yaml │ │ │ └── zinc_subset.yaml │ │ └── model │ │ │ ├── subset_128.yaml │ │ │ └── subset_32.yaml │ ├── data.py │ ├── dataset_statistics.py │ ├── pretrained_checkpoints.py │ ├── test_on_ogb.py │ ├── test_on_zinc.py │ ├── train_combo_on_ogb.py │ ├── train_combo_on_zinc.py │ ├── train_path_on_zinc.py │ └── utils.py │ ├── model.py │ ├── pathnet.py │ ├── statistics.py │ ├── transform.py │ └── utils.py └── tests └── test_blocks.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/.gitignore -------------------------------------------------------------------------------- /AUTHORS.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/AUTHORS.rst -------------------------------------------------------------------------------- /CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/CHANGELOG.rst -------------------------------------------------------------------------------- /CONTRIBUTING.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/CONTRIBUTING.rst -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/README.md -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/setup.py -------------------------------------------------------------------------------- /single_layer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/single_layer.png -------------------------------------------------------------------------------- /src/autobahn/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.0.0' 2 | -------------------------------------------------------------------------------- /src/autobahn/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/blocks.py -------------------------------------------------------------------------------- /src/autobahn/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/datasets.py -------------------------------------------------------------------------------- /src/autobahn/decompositions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/decompositions.py -------------------------------------------------------------------------------- /src/autobahn/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | """Runs code for experiments""" 2 | -------------------------------------------------------------------------------- /src/autobahn/experiments/combo_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/combo_models.py -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/config_ogb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/config_ogb.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/config_zinc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/config_zinc.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/ogbg_hiv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/ogbg_hiv.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/ogbg_molpcba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/ogbg_molpcba.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/ogbg_muv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/ogbg_muv.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/ogbg_tox.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/ogbg_tox.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/zinc_full.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/zinc_full.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/dataset/zinc_subset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/dataset/zinc_subset.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/model/subset_128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/model/subset_128.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/conf/model/subset_32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/conf/model/subset_32.yaml -------------------------------------------------------------------------------- /src/autobahn/experiments/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/data.py -------------------------------------------------------------------------------- /src/autobahn/experiments/dataset_statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/dataset_statistics.py -------------------------------------------------------------------------------- /src/autobahn/experiments/pretrained_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/pretrained_checkpoints.py -------------------------------------------------------------------------------- /src/autobahn/experiments/test_on_ogb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/test_on_ogb.py -------------------------------------------------------------------------------- /src/autobahn/experiments/test_on_zinc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/test_on_zinc.py -------------------------------------------------------------------------------- /src/autobahn/experiments/train_combo_on_ogb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/train_combo_on_ogb.py -------------------------------------------------------------------------------- /src/autobahn/experiments/train_combo_on_zinc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/train_combo_on_zinc.py -------------------------------------------------------------------------------- /src/autobahn/experiments/train_path_on_zinc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/train_path_on_zinc.py -------------------------------------------------------------------------------- /src/autobahn/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/experiments/utils.py -------------------------------------------------------------------------------- /src/autobahn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/model.py -------------------------------------------------------------------------------- /src/autobahn/pathnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/pathnet.py -------------------------------------------------------------------------------- /src/autobahn/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/statistics.py -------------------------------------------------------------------------------- /src/autobahn/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/transform.py -------------------------------------------------------------------------------- /src/autobahn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/src/autobahn/utils.py -------------------------------------------------------------------------------- /tests/test_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/risilab/Autobahn/HEAD/tests/test_blocks.py --------------------------------------------------------------------------------