文件名称：HGCAE:HGCAE Pytorch实施。 接受CVPR2021
文件大小：17.28MB
文件格式：ZIP
更新时间：2024-04-20 03:53:31
Python
双曲图卷积自动编码器 接受CVPR2021 :party_popper: 的官方PyTorch代码 *， *，Hyung Jin Chang，Jin Young Choi （*表示同等贡献） cora数据集的嵌入。 是欧氏空间中的图形自动编码器，而HGCAE是我们的方法。 P是庞加莱球，H是双曲面。 概述 该存储库在PyTorch中提供了HGCAE代码，可通过 PoincareBall歧管 图数据的链接预测任务和节点聚类任务 6个数据集： Cora, Citeseer, Wiki, Pubmed, Blog Catalog, Amazon Photo Amazon Photo是通过torch-geometric包装下载的。 图像上的图像聚类任务 2个数据集： ImageNet10, ImageNetDog 使用图像聚类算法从ImageNet10, ImageNetDog提取图像特征 来自提供的图像特
【文件预览】：
HGCAE-master
----models()
--------base_models.py(9KB)
--------encoders.py(6KB)
--------__init__.py(0B)
--------decoders.py(7KB)
----images()
--------fig_cora_small.png(363KB)
--------tree_BNCR.png(233KB)
----optimizers()
--------radam.py(7KB)
--------__init__.py(63B)
----data()
--------PICA-ImageNetDog()
--------ImageNet-BNCR()
--------pubmed()
--------wiki()
--------PICA-ImageNet10()
--------citeseer()
--------BlogCatalog()
--------cora()
----manifolds()
--------euclidean.py(1KB)
--------__init__.py(211B)
--------poincare.py(5KB)
--------base.py(2KB)
----LICENSE(1KB)
----__init__.py(70B)
----solver.py(9KB)
----layers()
--------hyp_layers.py(8KB)
--------att_layers.py(3KB)
--------__init__.py(0B)
--------layers.py(3KB)
----.gitignore(12B)
----config.py(4KB)
----README.md(5KB)
----script()
--------train_wiki_nc.sh(878B)
--------train_cora_lp.sh(850B)
--------train_pubmed_lp.sh(859B)
--------train_pubmed_nc.sh(874B)
--------train_blogcatalog_lp.sh(855B)
--------train_citeseer_nc.sh(883B)
--------train_wiki_lp.sh(859B)
--------train_ImageNet10.sh(981B)
--------train_cora_nc.sh(873B)
--------train_blogcatalog_nc.sh(876B)
--------train_citeseer_lp.sh(853B)
--------train_amazonphoto_nc.sh(880B)
--------train_ImageNetDog.sh(982B)
--------train_amazonphoto_lp.sh(863B)
----utils()
--------train_utils.py(4KB)
--------math_utils.py(2KB)
--------eval_utils.py(370B)
--------__init__.py(0B)
--------data_utils.py(15KB)
----guide.md(3KB)
----train_solver.py(2KB)