├── .gitignore ├── LICENSE ├── README.md ├── config ├── dataset │ ├── cifar10.yaml │ ├── cluster.yaml │ ├── coco.yaml │ ├── mnist.yaml │ ├── ogbg_code2.yaml │ ├── ogbg_molpcba.yaml │ ├── ogbg_ppa.yaml │ ├── pascalvoc.yaml │ ├── pattern.yaml │ ├── pcqm_contact.yaml │ ├── peptides_func.yaml │ ├── peptides_struct.yaml │ └── zinc.yaml ├── experiment │ ├── cifar10.yaml │ ├── cluster.yaml │ ├── coco.yaml │ ├── mnist.yaml │ ├── model │ │ ├── conv+vn_3L.yaml │ │ ├── mamba+transformer_5L.yaml │ │ ├── mamba+vn_3L.yaml │ │ ├── mamba+vn_6L.yaml │ │ ├── mamba+vn_cluster.yaml │ │ ├── mamba_3L.yaml │ │ ├── mamba_6L.yaml │ │ ├── ogbg_code2.yaml │ │ ├── ogbg_molpcba.yaml │ │ ├── ogbg_ppa.yaml │ │ ├── s4+vn_3L.yaml │ │ └── transformer+vn_3L.yaml │ ├── ogbg_code2.yaml │ ├── ogbg_molpcba.yaml │ ├── ogbg_ppa.yaml │ ├── pascalvoc.yaml │ ├── pattern.yaml │ ├── pcqm_contact.yaml │ ├── peptides_func.yaml │ ├── peptides_struct.yaml │ └── zinc.yaml ├── mode │ ├── debug.yaml │ └── default.yaml ├── results.yaml └── train.yaml ├── environment.yaml ├── environment_latest.yaml ├── images └── overview.png ├── neuralwalker ├── data │ ├── __init__.py │ ├── ogbg_code2_utils.py │ ├── transforms.py │ ├── walks.pyx │ └── wrapper.py ├── loss.py ├── lr_schedulers.py ├── models │ ├── __init__.py │ ├── neuralwalker.py │ └── pl_models.py ├── modules │ ├── __init__.py │ ├── feature_encoder.py │ ├── gnn_layers.py │ ├── head.py │ ├── mp_layer.py │ ├── neuralwalker_layer.py │ ├── pos_embed.py │ ├── rotary_embedding.py │ ├── s4.py │ ├── transformer.py │ └── virtual_node_layer.py └── utils.py ├── node_classification ├── LICENSE ├── README.md ├── data_utils.py ├── dataset.py ├── eval.py ├── large_graph_exp │ ├── data │ │ └── pokec │ │ │ └── pokec-splits.npy │ ├── lg_model.py │ ├── lg_parse.py │ ├── main-batch.py │ ├── run_large.sh │ └── walk_encoder.py ├── logger.py ├── main.py ├── model.py ├── parse.py ├── run.sh ├── utils.py └── walk_encoder.py ├── pyproject.toml ├── scripts └── get_results.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/README.md -------------------------------------------------------------------------------- /config/dataset/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/cifar10.yaml -------------------------------------------------------------------------------- /config/dataset/cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/cluster.yaml -------------------------------------------------------------------------------- /config/dataset/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/coco.yaml -------------------------------------------------------------------------------- /config/dataset/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/mnist.yaml -------------------------------------------------------------------------------- /config/dataset/ogbg_code2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/ogbg_code2.yaml -------------------------------------------------------------------------------- /config/dataset/ogbg_molpcba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/ogbg_molpcba.yaml -------------------------------------------------------------------------------- /config/dataset/ogbg_ppa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/ogbg_ppa.yaml -------------------------------------------------------------------------------- /config/dataset/pascalvoc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/pascalvoc.yaml -------------------------------------------------------------------------------- /config/dataset/pattern.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/pattern.yaml -------------------------------------------------------------------------------- /config/dataset/pcqm_contact.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/pcqm_contact.yaml -------------------------------------------------------------------------------- /config/dataset/peptides_func.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/peptides_func.yaml -------------------------------------------------------------------------------- /config/dataset/peptides_struct.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/peptides_struct.yaml -------------------------------------------------------------------------------- /config/dataset/zinc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/dataset/zinc.yaml -------------------------------------------------------------------------------- /config/experiment/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/cifar10.yaml -------------------------------------------------------------------------------- /config/experiment/cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/cluster.yaml -------------------------------------------------------------------------------- /config/experiment/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/coco.yaml -------------------------------------------------------------------------------- /config/experiment/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/mnist.yaml -------------------------------------------------------------------------------- /config/experiment/model/conv+vn_3L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/conv+vn_3L.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba+transformer_5L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba+transformer_5L.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba+vn_3L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba+vn_3L.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba+vn_6L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba+vn_6L.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba+vn_cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba+vn_cluster.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba_3L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba_3L.yaml -------------------------------------------------------------------------------- /config/experiment/model/mamba_6L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/mamba_6L.yaml -------------------------------------------------------------------------------- /config/experiment/model/ogbg_code2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/ogbg_code2.yaml -------------------------------------------------------------------------------- /config/experiment/model/ogbg_molpcba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/ogbg_molpcba.yaml -------------------------------------------------------------------------------- /config/experiment/model/ogbg_ppa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/ogbg_ppa.yaml -------------------------------------------------------------------------------- /config/experiment/model/s4+vn_3L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/s4+vn_3L.yaml -------------------------------------------------------------------------------- /config/experiment/model/transformer+vn_3L.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/model/transformer+vn_3L.yaml -------------------------------------------------------------------------------- /config/experiment/ogbg_code2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/ogbg_code2.yaml -------------------------------------------------------------------------------- /config/experiment/ogbg_molpcba.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/ogbg_molpcba.yaml -------------------------------------------------------------------------------- /config/experiment/ogbg_ppa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/ogbg_ppa.yaml -------------------------------------------------------------------------------- /config/experiment/pascalvoc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/pascalvoc.yaml -------------------------------------------------------------------------------- /config/experiment/pattern.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/pattern.yaml -------------------------------------------------------------------------------- /config/experiment/pcqm_contact.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/pcqm_contact.yaml -------------------------------------------------------------------------------- /config/experiment/peptides_func.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/peptides_func.yaml -------------------------------------------------------------------------------- /config/experiment/peptides_struct.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/peptides_struct.yaml -------------------------------------------------------------------------------- /config/experiment/zinc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/experiment/zinc.yaml -------------------------------------------------------------------------------- /config/mode/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/mode/debug.yaml -------------------------------------------------------------------------------- /config/mode/default.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/results.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/results.yaml -------------------------------------------------------------------------------- /config/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/config/train.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/environment.yaml -------------------------------------------------------------------------------- /environment_latest.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/environment_latest.yaml -------------------------------------------------------------------------------- /images/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/images/overview.png -------------------------------------------------------------------------------- /neuralwalker/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/data/__init__.py -------------------------------------------------------------------------------- /neuralwalker/data/ogbg_code2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/data/ogbg_code2_utils.py -------------------------------------------------------------------------------- /neuralwalker/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/data/transforms.py -------------------------------------------------------------------------------- /neuralwalker/data/walks.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/data/walks.pyx -------------------------------------------------------------------------------- /neuralwalker/data/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/data/wrapper.py -------------------------------------------------------------------------------- /neuralwalker/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/loss.py -------------------------------------------------------------------------------- /neuralwalker/lr_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/lr_schedulers.py -------------------------------------------------------------------------------- /neuralwalker/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/models/__init__.py -------------------------------------------------------------------------------- /neuralwalker/models/neuralwalker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/models/neuralwalker.py -------------------------------------------------------------------------------- /neuralwalker/models/pl_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/models/pl_models.py -------------------------------------------------------------------------------- /neuralwalker/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralwalker/modules/feature_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/feature_encoder.py -------------------------------------------------------------------------------- /neuralwalker/modules/gnn_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/gnn_layers.py -------------------------------------------------------------------------------- /neuralwalker/modules/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/head.py -------------------------------------------------------------------------------- /neuralwalker/modules/mp_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/mp_layer.py -------------------------------------------------------------------------------- /neuralwalker/modules/neuralwalker_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/neuralwalker_layer.py -------------------------------------------------------------------------------- /neuralwalker/modules/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/pos_embed.py -------------------------------------------------------------------------------- /neuralwalker/modules/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/rotary_embedding.py -------------------------------------------------------------------------------- /neuralwalker/modules/s4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/s4.py -------------------------------------------------------------------------------- /neuralwalker/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/transformer.py -------------------------------------------------------------------------------- /neuralwalker/modules/virtual_node_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/modules/virtual_node_layer.py -------------------------------------------------------------------------------- /neuralwalker/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/neuralwalker/utils.py -------------------------------------------------------------------------------- /node_classification/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/LICENSE -------------------------------------------------------------------------------- /node_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/README.md -------------------------------------------------------------------------------- /node_classification/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/data_utils.py -------------------------------------------------------------------------------- /node_classification/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/dataset.py -------------------------------------------------------------------------------- /node_classification/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/eval.py -------------------------------------------------------------------------------- /node_classification/large_graph_exp/data/pokec/pokec-splits.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/data/pokec/pokec-splits.npy -------------------------------------------------------------------------------- /node_classification/large_graph_exp/lg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/lg_model.py -------------------------------------------------------------------------------- /node_classification/large_graph_exp/lg_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/lg_parse.py -------------------------------------------------------------------------------- /node_classification/large_graph_exp/main-batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/main-batch.py -------------------------------------------------------------------------------- /node_classification/large_graph_exp/run_large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/run_large.sh -------------------------------------------------------------------------------- /node_classification/large_graph_exp/walk_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/large_graph_exp/walk_encoder.py -------------------------------------------------------------------------------- /node_classification/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/logger.py -------------------------------------------------------------------------------- /node_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/main.py -------------------------------------------------------------------------------- /node_classification/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/model.py -------------------------------------------------------------------------------- /node_classification/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/parse.py -------------------------------------------------------------------------------- /node_classification/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/run.sh -------------------------------------------------------------------------------- /node_classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/utils.py -------------------------------------------------------------------------------- /node_classification/walk_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/node_classification/walk_encoder.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/get_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/scripts/get_results.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BorgwardtLab/NeuralWalker/HEAD/train.py --------------------------------------------------------------------------------