├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── __init__.py ├── data ├── ethereum │ └── download_ethereum.sh ├── node │ └── download_node.sh ├── qm8 │ ├── get_data.py │ ├── test_idx_qm8.json │ ├── utils.py │ └── valid_idx_qm8.json ├── qm9 │ ├── get_data.py │ ├── test_idx_qm9.json │ ├── utils.py │ └── valid_idx_qm9.json ├── synthetic │ └── generate_graphs.py ├── tu │ └── data_preprocess.py └── zinc │ ├── get_data.py │ ├── test_idx_zinc.json │ ├── utils.py │ └── valid_idx_zinc.json ├── dataset ├── EthereumDataset.py ├── GraphDataset.py ├── NodeClassificationDataset.py ├── SyntheticDataset.py └── __init__.py ├── gnn ├── RiemannianGNN.py └── __init__.py ├── hyperbolic_module ├── CentroidDistance.py ├── PoincareDistance.py └── __init__.py ├── install.sh ├── main.py ├── manifold ├── EuclideanManifold.py ├── LorentzManifold.py ├── PoincareManifold.py └── __init__.py ├── optimizer ├── __init__.py ├── ramsgrad.py └── rsgd.py ├── params ├── CollabEuclideanParams.py ├── CollabHyperbolicParams.py ├── DDEuclideanParams.py ├── DDHyperbolicParams.py ├── EnzymesEuclideanParams.py ├── EnzymesHyperbolicParams.py ├── EthereumEuclideanParams.py ├── EthereumHyperbolicParams.py ├── NodeClassificationHyperbolicParams.py ├── ProteinsEuclideanParams.py ├── ProteinsHyperbolicParams.py ├── QM8EuclideanParams.py ├── QM8HyperbolicParams.py ├── QM9EuclideanParams.py ├── QM9HyperbolicParams.py ├── RedditEuclideanParams.py ├── RedditHyperbolicParams.py ├── SyntheticEuclideanParams.py ├── SyntheticHyperbolicParams.py ├── ZINCEuclideanParams.py ├── ZINCHyperbolicParams.py └── __init__.py ├── requirements.txt ├── task ├── BaseTask.py ├── GraphPrediction.py ├── GraphPredictionTask.py ├── GraphSeriesPrediction.py ├── GraphSeriesPredictionTask.py ├── NodeClassification.py ├── NodeClassificationTask.py └── __init__.py └── utils ├── EarlyStoppingCriterion.py ├── __init__.py ├── logger.py └── utils.py /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/ethereum/download_ethereum.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/ethereum/download_ethereum.sh -------------------------------------------------------------------------------- /data/node/download_node.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/node/download_node.sh -------------------------------------------------------------------------------- /data/qm8/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm8/get_data.py -------------------------------------------------------------------------------- /data/qm8/test_idx_qm8.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm8/test_idx_qm8.json -------------------------------------------------------------------------------- /data/qm8/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm8/utils.py -------------------------------------------------------------------------------- /data/qm8/valid_idx_qm8.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm8/valid_idx_qm8.json -------------------------------------------------------------------------------- /data/qm9/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm9/get_data.py -------------------------------------------------------------------------------- /data/qm9/test_idx_qm9.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm9/test_idx_qm9.json -------------------------------------------------------------------------------- /data/qm9/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm9/utils.py -------------------------------------------------------------------------------- /data/qm9/valid_idx_qm9.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/qm9/valid_idx_qm9.json -------------------------------------------------------------------------------- /data/synthetic/generate_graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/synthetic/generate_graphs.py -------------------------------------------------------------------------------- /data/tu/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/tu/data_preprocess.py -------------------------------------------------------------------------------- /data/zinc/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/zinc/get_data.py -------------------------------------------------------------------------------- /data/zinc/test_idx_zinc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/zinc/test_idx_zinc.json -------------------------------------------------------------------------------- /data/zinc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/zinc/utils.py -------------------------------------------------------------------------------- /data/zinc/valid_idx_zinc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/data/zinc/valid_idx_zinc.json -------------------------------------------------------------------------------- /dataset/EthereumDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/dataset/EthereumDataset.py -------------------------------------------------------------------------------- /dataset/GraphDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/dataset/GraphDataset.py -------------------------------------------------------------------------------- /dataset/NodeClassificationDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/dataset/NodeClassificationDataset.py -------------------------------------------------------------------------------- /dataset/SyntheticDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/dataset/SyntheticDataset.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /gnn/RiemannianGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/gnn/RiemannianGNN.py -------------------------------------------------------------------------------- /gnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/gnn/__init__.py -------------------------------------------------------------------------------- /hyperbolic_module/CentroidDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/hyperbolic_module/CentroidDistance.py -------------------------------------------------------------------------------- /hyperbolic_module/PoincareDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/hyperbolic_module/PoincareDistance.py -------------------------------------------------------------------------------- /hyperbolic_module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/hyperbolic_module/__init__.py -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/install.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/main.py -------------------------------------------------------------------------------- /manifold/EuclideanManifold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/manifold/EuclideanManifold.py -------------------------------------------------------------------------------- /manifold/LorentzManifold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/manifold/LorentzManifold.py -------------------------------------------------------------------------------- /manifold/PoincareManifold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/manifold/PoincareManifold.py -------------------------------------------------------------------------------- /manifold/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/manifold/__init__.py -------------------------------------------------------------------------------- /optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/optimizer/__init__.py -------------------------------------------------------------------------------- /optimizer/ramsgrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/optimizer/ramsgrad.py -------------------------------------------------------------------------------- /optimizer/rsgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/optimizer/rsgd.py -------------------------------------------------------------------------------- /params/CollabEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/CollabEuclideanParams.py -------------------------------------------------------------------------------- /params/CollabHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/CollabHyperbolicParams.py -------------------------------------------------------------------------------- /params/DDEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/DDEuclideanParams.py -------------------------------------------------------------------------------- /params/DDHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/DDHyperbolicParams.py -------------------------------------------------------------------------------- /params/EnzymesEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/EnzymesEuclideanParams.py -------------------------------------------------------------------------------- /params/EnzymesHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/EnzymesHyperbolicParams.py -------------------------------------------------------------------------------- /params/EthereumEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/EthereumEuclideanParams.py -------------------------------------------------------------------------------- /params/EthereumHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/EthereumHyperbolicParams.py -------------------------------------------------------------------------------- /params/NodeClassificationHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/NodeClassificationHyperbolicParams.py -------------------------------------------------------------------------------- /params/ProteinsEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/ProteinsEuclideanParams.py -------------------------------------------------------------------------------- /params/ProteinsHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/ProteinsHyperbolicParams.py -------------------------------------------------------------------------------- /params/QM8EuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/QM8EuclideanParams.py -------------------------------------------------------------------------------- /params/QM8HyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/QM8HyperbolicParams.py -------------------------------------------------------------------------------- /params/QM9EuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/QM9EuclideanParams.py -------------------------------------------------------------------------------- /params/QM9HyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/QM9HyperbolicParams.py -------------------------------------------------------------------------------- /params/RedditEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/RedditEuclideanParams.py -------------------------------------------------------------------------------- /params/RedditHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/RedditHyperbolicParams.py -------------------------------------------------------------------------------- /params/SyntheticEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/SyntheticEuclideanParams.py -------------------------------------------------------------------------------- /params/SyntheticHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/SyntheticHyperbolicParams.py -------------------------------------------------------------------------------- /params/ZINCEuclideanParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/ZINCEuclideanParams.py -------------------------------------------------------------------------------- /params/ZINCHyperbolicParams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/ZINCHyperbolicParams.py -------------------------------------------------------------------------------- /params/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/params/__init__.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/requirements.txt -------------------------------------------------------------------------------- /task/BaseTask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/BaseTask.py -------------------------------------------------------------------------------- /task/GraphPrediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/GraphPrediction.py -------------------------------------------------------------------------------- /task/GraphPredictionTask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/GraphPredictionTask.py -------------------------------------------------------------------------------- /task/GraphSeriesPrediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/GraphSeriesPrediction.py -------------------------------------------------------------------------------- /task/GraphSeriesPredictionTask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/GraphSeriesPredictionTask.py -------------------------------------------------------------------------------- /task/NodeClassification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/NodeClassification.py -------------------------------------------------------------------------------- /task/NodeClassificationTask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/NodeClassificationTask.py -------------------------------------------------------------------------------- /task/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/task/__init__.py -------------------------------------------------------------------------------- /utils/EarlyStoppingCriterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/utils/EarlyStoppingCriterion.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/hgnn/HEAD/utils/utils.py --------------------------------------------------------------------------------