├── .gitignore ├── LICENSE ├── README.md ├── data ├── __init__.py ├── datasets.py ├── loader.py └── mappings.py ├── exp ├── __init__.py ├── pretrain.py ├── supervised.py └── transfer.py ├── main.py ├── manifolds ├── __init__.py ├── euclidean.py ├── lorentz.py ├── product.py └── sphere.py ├── model.png ├── modules ├── __init__.py ├── basics.py ├── heads.py ├── layers.py └── model.py ├── requirements.txt ├── results_1.png ├── results_2.png ├── scripts ├── 1-shot_NC │ ├── citeseer.sh │ ├── github.sh │ ├── pubmed.sh │ └── usa.sh ├── 5-shot_NC │ ├── citeseer.sh │ ├── github.sh │ ├── pubmed.sh │ └── usa.sh ├── LP │ ├── citeseer.sh │ ├── cora.sh │ ├── github.sh │ ├── pubmed.sh │ └── usa.sh ├── NC │ ├── citeseer.sh │ ├── github.sh │ ├── pubmed.sh │ └── usa.sh ├── finetune.sh └── pretrain.sh └── utils ├── __init__.py ├── config.py ├── evall_utils.py ├── logger.py ├── math_utils.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/data/datasets.py -------------------------------------------------------------------------------- /data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/data/loader.py -------------------------------------------------------------------------------- /data/mappings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/data/mappings.py -------------------------------------------------------------------------------- /exp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/exp/__init__.py -------------------------------------------------------------------------------- /exp/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/exp/pretrain.py -------------------------------------------------------------------------------- /exp/supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/exp/supervised.py -------------------------------------------------------------------------------- /exp/transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/exp/transfer.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/main.py -------------------------------------------------------------------------------- /manifolds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/manifolds/__init__.py -------------------------------------------------------------------------------- /manifolds/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/manifolds/euclidean.py -------------------------------------------------------------------------------- /manifolds/lorentz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/manifolds/lorentz.py -------------------------------------------------------------------------------- /manifolds/product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/manifolds/product.py -------------------------------------------------------------------------------- /manifolds/sphere.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/manifolds/sphere.py -------------------------------------------------------------------------------- /model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/model.png -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/modules/basics.py -------------------------------------------------------------------------------- /modules/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/modules/heads.py -------------------------------------------------------------------------------- /modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/modules/layers.py -------------------------------------------------------------------------------- /modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/modules/model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/requirements.txt -------------------------------------------------------------------------------- /results_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/results_1.png -------------------------------------------------------------------------------- /results_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/results_2.png -------------------------------------------------------------------------------- /scripts/1-shot_NC/citeseer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/1-shot_NC/citeseer.sh -------------------------------------------------------------------------------- /scripts/1-shot_NC/github.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/1-shot_NC/github.sh -------------------------------------------------------------------------------- /scripts/1-shot_NC/pubmed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/1-shot_NC/pubmed.sh -------------------------------------------------------------------------------- /scripts/1-shot_NC/usa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/1-shot_NC/usa.sh -------------------------------------------------------------------------------- /scripts/5-shot_NC/citeseer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/5-shot_NC/citeseer.sh -------------------------------------------------------------------------------- /scripts/5-shot_NC/github.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/5-shot_NC/github.sh -------------------------------------------------------------------------------- /scripts/5-shot_NC/pubmed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/5-shot_NC/pubmed.sh -------------------------------------------------------------------------------- /scripts/5-shot_NC/usa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/5-shot_NC/usa.sh -------------------------------------------------------------------------------- /scripts/LP/citeseer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/LP/citeseer.sh -------------------------------------------------------------------------------- /scripts/LP/cora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/LP/cora.sh -------------------------------------------------------------------------------- /scripts/LP/github.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/LP/github.sh -------------------------------------------------------------------------------- /scripts/LP/pubmed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/LP/pubmed.sh -------------------------------------------------------------------------------- /scripts/LP/usa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/LP/usa.sh -------------------------------------------------------------------------------- /scripts/NC/citeseer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/NC/citeseer.sh -------------------------------------------------------------------------------- /scripts/NC/github.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/NC/github.sh -------------------------------------------------------------------------------- /scripts/NC/pubmed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/NC/pubmed.sh -------------------------------------------------------------------------------- /scripts/NC/usa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/NC/usa.sh -------------------------------------------------------------------------------- /scripts/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/finetune.sh -------------------------------------------------------------------------------- /scripts/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/scripts/pretrain.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/evall_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/utils/evall_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/utils/math_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RiemannGraph/RiemannGFM/HEAD/utils/train_utils.py --------------------------------------------------------------------------------