├── LICENSE.md ├── README.md ├── bgrl ├── __init__.py ├── bgrl.py ├── data.py ├── linear_eval_ppi.py ├── logistic_regression_eval.py ├── models.py ├── predictors.py ├── scheduler.py ├── transforms.py └── utils.py ├── bgrl_arch.png ├── config-eval ├── amazon-computers.cfg ├── amazon-photos.cfg ├── coauthor-cs.cfg ├── coauthor-physics.cfg ├── ppi.cfg └── wiki-cs.cfg ├── config ├── amazon-computers.cfg ├── amazon-photos.cfg ├── coauthor-cs.cfg ├── coauthor-physics.cfg ├── ppi.cfg └── wiki-cs.cfg ├── linear_eval_ppi.py ├── linear_eval_transductive.py ├── train_ppi.py ├── train_transductive.py └── weights ├── bgrl-amazon-computers.pt ├── bgrl-amazon-photos.pt ├── bgrl-coauthor-cs.pt ├── bgrl-coauthor-physics.pt ├── bgrl-ppi.pt └── bgrl-wikics.pt /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/README.md -------------------------------------------------------------------------------- /bgrl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/__init__.py -------------------------------------------------------------------------------- /bgrl/bgrl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/bgrl.py -------------------------------------------------------------------------------- /bgrl/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/data.py -------------------------------------------------------------------------------- /bgrl/linear_eval_ppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/linear_eval_ppi.py -------------------------------------------------------------------------------- /bgrl/logistic_regression_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/logistic_regression_eval.py -------------------------------------------------------------------------------- /bgrl/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/models.py -------------------------------------------------------------------------------- /bgrl/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/predictors.py -------------------------------------------------------------------------------- /bgrl/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/scheduler.py -------------------------------------------------------------------------------- /bgrl/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/transforms.py -------------------------------------------------------------------------------- /bgrl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl/utils.py -------------------------------------------------------------------------------- /bgrl_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/bgrl_arch.png -------------------------------------------------------------------------------- /config-eval/amazon-computers.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/amazon-computers.cfg -------------------------------------------------------------------------------- /config-eval/amazon-photos.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/amazon-photos.cfg -------------------------------------------------------------------------------- /config-eval/coauthor-cs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/coauthor-cs.cfg -------------------------------------------------------------------------------- /config-eval/coauthor-physics.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/coauthor-physics.cfg -------------------------------------------------------------------------------- /config-eval/ppi.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/ppi.cfg -------------------------------------------------------------------------------- /config-eval/wiki-cs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config-eval/wiki-cs.cfg -------------------------------------------------------------------------------- /config/amazon-computers.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/amazon-computers.cfg -------------------------------------------------------------------------------- /config/amazon-photos.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/amazon-photos.cfg -------------------------------------------------------------------------------- /config/coauthor-cs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/coauthor-cs.cfg -------------------------------------------------------------------------------- /config/coauthor-physics.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/coauthor-physics.cfg -------------------------------------------------------------------------------- /config/ppi.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/ppi.cfg -------------------------------------------------------------------------------- /config/wiki-cs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/config/wiki-cs.cfg -------------------------------------------------------------------------------- /linear_eval_ppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/linear_eval_ppi.py -------------------------------------------------------------------------------- /linear_eval_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/linear_eval_transductive.py -------------------------------------------------------------------------------- /train_ppi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/train_ppi.py -------------------------------------------------------------------------------- /train_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/train_transductive.py -------------------------------------------------------------------------------- /weights/bgrl-amazon-computers.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-amazon-computers.pt -------------------------------------------------------------------------------- /weights/bgrl-amazon-photos.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-amazon-photos.pt -------------------------------------------------------------------------------- /weights/bgrl-coauthor-cs.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-coauthor-cs.pt -------------------------------------------------------------------------------- /weights/bgrl-coauthor-physics.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-coauthor-physics.pt -------------------------------------------------------------------------------- /weights/bgrl-ppi.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-ppi.pt -------------------------------------------------------------------------------- /weights/bgrl-wikics.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nerdslab/bgrl/HEAD/weights/bgrl-wikics.pt --------------------------------------------------------------------------------