【文件属性】：
文件名称：hgnn:双曲图神经网络
文件大小：271KB
文件格式：ZIP
更新时间：2021-04-16 04:01:08
Python
双曲图神经网络
要求
Python 3.7
PyTorch> = 1.1
RDKit
麻木
网络
scikit学习
install.sh提供了有关安装需求的食谱。
数据预处理
对于以太坊数据集，转到data/ethereum并运行download_ethereum.sh
对于节点分类数据集，请转到data/node并运行download_node.sh
对于QM8，QM9和ZINC，分别转到data/qm8 ， data/qm9和data/zinc并运行python get_data.py
对于合成数据集，转到data/synthetic并运行python generate_graphs.py
对于多特蒙德python data_preprocess.py {REDDIT-MULTI-12K, PROTEINS_full, ENZYMES, DD, COLLAB}大学的数据集
【文件预览】：
hgnn-master
----utils()
--------utils.py(8KB)
--------__init__.py(336B)
--------EarlyStoppingCriterion.py(2KB)
--------logger.py(2KB)
----install.sh(885B)
----main.py(6KB)
----requirements.txt(45B)
----__init__.py(0B)
----dataset()
--------NodeClassificationDataset.py(6KB)
--------GraphDataset.py(2KB)
--------__init__.py(218B)
--------SyntheticDataset.py(2KB)
--------EthereumDataset.py(6KB)
----CONTRIBUTING.md(1KB)
----LICENSE(19KB)
----README.md(3KB)
----params()
--------NodeClassificationHyperbolicParams.py(2KB)
--------ZINCEuclideanParams.py(3KB)
--------EthereumEuclideanParams.py(3KB)
--------CollabEuclideanParams.py(3KB)
--------ZINCHyperbolicParams.py(3KB)
--------ProteinsEuclideanParams.py(3KB)
--------QM8HyperbolicParams.py(4KB)
--------__init__.py(990B)
--------SyntheticEuclideanParams.py(3KB)
--------SyntheticHyperbolicParams.py(3KB)
--------RedditEuclideanParams.py(3KB)
--------RedditHyperbolicParams.py(3KB)
--------DDHyperbolicParams.py(3KB)
--------QM8EuclideanParams.py(3KB)
--------QM9EuclideanParams.py(3KB)
--------EthereumHyperbolicParams.py(3KB)
--------EnzymesHyperbolicParams.py(3KB)
--------CollabHyperbolicParams.py(3KB)
--------DDEuclideanParams.py(3KB)
--------EnzymesEuclideanParams.py(3KB)
--------ProteinsHyperbolicParams.py(3KB)
--------QM9HyperbolicParams.py(4KB)
----gnn()
--------RiemannianGNN.py(6KB)
--------__init__.py(263B)
----data()
--------zinc()
--------synthetic()
--------node()
--------qm8()
--------tu()
--------qm9()
--------ethereum()
----hyperbolic_module()
--------PoincareDistance.py(2KB)
--------CentroidDistance.py(2KB)
--------__init__.py(219B)
----manifold()
--------LorentzManifold.py(5KB)
--------__init__.py(339B)
--------EuclideanManifold.py(1KB)
--------PoincareManifold.py(4KB)
----CODE_OF_CONDUCT.md(244B)
----task()
--------NodeClassification.py(2KB)
--------NodeClassificationTask.py(4KB)
--------__init__.py(345B)
--------BaseTask.py(4KB)
--------GraphSeriesPrediction.py(3KB)
--------GraphPredictionTask.py(5KB)
--------GraphPrediction.py(2KB)
--------GraphSeriesPredictionTask.py(4KB)
----optimizer()
--------ramsgrad.py(3KB)
--------rsgd.py(2KB)
--------__init__.py(218B)