├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── config.py ├── data ├── BlogCatalog │ └── BlogCatalog.mat ├── ImageNet-BNCR │ └── image_idx.txt ├── PICA-ImageNet10 │ ├── A-picafeat10_imagenet10-k20-mknn.npz │ ├── label.npy │ └── picafeat10_imagenet10.npy ├── PICA-ImageNetDog │ ├── A-picafeat10_imagenetdog-k10-mknn.npz │ ├── label.npy │ └── picafeat10_imagenetdog.npy ├── citeseer │ ├── ind.citeseer.allx │ ├── ind.citeseer.ally │ ├── ind.citeseer.graph │ ├── ind.citeseer.test.index │ ├── ind.citeseer.tx │ ├── ind.citeseer.ty │ ├── ind.citeseer.x │ └── ind.citeseer.y ├── cora │ ├── ind.cora.allx │ ├── ind.cora.ally │ ├── ind.cora.graph │ ├── ind.cora.test.index │ ├── ind.cora.tx │ ├── ind.cora.ty │ ├── ind.cora.x │ └── ind.cora.y ├── pubmed │ ├── ind.pubmed.allx │ ├── ind.pubmed.ally │ ├── ind.pubmed.graph │ ├── ind.pubmed.test.index │ ├── ind.pubmed.tx │ ├── ind.pubmed.ty │ ├── ind.pubmed.x │ └── ind.pubmed.y └── wiki │ ├── wiki_feature.txt │ ├── wiki_graph.txt │ └── wiki_group.txt ├── guide.md ├── images ├── fig_cora_small.png └── tree_BNCR.png ├── layers ├── __init__.py ├── att_layers.py ├── hyp_layers.py └── layers.py ├── manifolds ├── __init__.py ├── base.py ├── euclidean.py └── poincare.py ├── models ├── __init__.py ├── base_models.py ├── decoders.py └── encoders.py ├── optimizers ├── __init__.py └── radam.py ├── script ├── train_ImageNet10.sh ├── train_ImageNetDog.sh ├── train_amazonphoto_lp.sh ├── train_amazonphoto_nc.sh ├── train_blogcatalog_lp.sh ├── train_blogcatalog_nc.sh ├── train_citeseer_lp.sh ├── train_citeseer_nc.sh ├── train_cora_lp.sh ├── train_cora_nc.sh ├── train_pubmed_lp.sh ├── train_pubmed_nc.sh ├── train_wiki_lp.sh └── train_wiki_nc.sh ├── solver.py ├── train_solver.py └── utils ├── __init__.py ├── data_utils.py ├── eval_utils.py ├── math_utils.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/__init__.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/config.py -------------------------------------------------------------------------------- /data/BlogCatalog/BlogCatalog.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/BlogCatalog/BlogCatalog.mat -------------------------------------------------------------------------------- /data/ImageNet-BNCR/image_idx.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/ImageNet-BNCR/image_idx.txt -------------------------------------------------------------------------------- /data/PICA-ImageNet10/A-picafeat10_imagenet10-k20-mknn.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNet10/A-picafeat10_imagenet10-k20-mknn.npz -------------------------------------------------------------------------------- /data/PICA-ImageNet10/label.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNet10/label.npy -------------------------------------------------------------------------------- /data/PICA-ImageNet10/picafeat10_imagenet10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNet10/picafeat10_imagenet10.npy -------------------------------------------------------------------------------- /data/PICA-ImageNetDog/A-picafeat10_imagenetdog-k10-mknn.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNetDog/A-picafeat10_imagenetdog-k10-mknn.npz -------------------------------------------------------------------------------- /data/PICA-ImageNetDog/label.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNetDog/label.npy -------------------------------------------------------------------------------- /data/PICA-ImageNetDog/picafeat10_imagenetdog.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/PICA-ImageNetDog/picafeat10_imagenetdog.npy -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.allx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.allx -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.ally: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.ally -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.graph -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.test.index -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.tx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.tx -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.ty: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.ty -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.x: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.x -------------------------------------------------------------------------------- /data/citeseer/ind.citeseer.y: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/citeseer/ind.citeseer.y -------------------------------------------------------------------------------- /data/cora/ind.cora.allx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.allx -------------------------------------------------------------------------------- /data/cora/ind.cora.ally: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.ally -------------------------------------------------------------------------------- /data/cora/ind.cora.graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.graph -------------------------------------------------------------------------------- /data/cora/ind.cora.test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.test.index -------------------------------------------------------------------------------- /data/cora/ind.cora.tx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.tx -------------------------------------------------------------------------------- /data/cora/ind.cora.ty: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.ty -------------------------------------------------------------------------------- /data/cora/ind.cora.x: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.x -------------------------------------------------------------------------------- /data/cora/ind.cora.y: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/cora/ind.cora.y -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.allx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.allx -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.ally: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.ally -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.graph -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.test.index -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.tx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.tx -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.ty: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.ty -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.x: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.x -------------------------------------------------------------------------------- /data/pubmed/ind.pubmed.y: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/pubmed/ind.pubmed.y -------------------------------------------------------------------------------- /data/wiki/wiki_feature.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/wiki/wiki_feature.txt -------------------------------------------------------------------------------- /data/wiki/wiki_graph.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/wiki/wiki_graph.txt -------------------------------------------------------------------------------- /data/wiki/wiki_group.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/data/wiki/wiki_group.txt -------------------------------------------------------------------------------- /guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/guide.md -------------------------------------------------------------------------------- /images/fig_cora_small.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/images/fig_cora_small.png -------------------------------------------------------------------------------- /images/tree_BNCR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/images/tree_BNCR.png -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/att_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/layers/att_layers.py -------------------------------------------------------------------------------- /layers/hyp_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/layers/hyp_layers.py -------------------------------------------------------------------------------- /layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/layers/layers.py -------------------------------------------------------------------------------- /manifolds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/manifolds/__init__.py -------------------------------------------------------------------------------- /manifolds/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/manifolds/base.py -------------------------------------------------------------------------------- /manifolds/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/manifolds/euclidean.py -------------------------------------------------------------------------------- /manifolds/poincare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/manifolds/poincare.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/models/base_models.py -------------------------------------------------------------------------------- /models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/models/decoders.py -------------------------------------------------------------------------------- /models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/models/encoders.py -------------------------------------------------------------------------------- /optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/optimizers/__init__.py -------------------------------------------------------------------------------- /optimizers/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/optimizers/radam.py -------------------------------------------------------------------------------- /script/train_ImageNet10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_ImageNet10.sh -------------------------------------------------------------------------------- /script/train_ImageNetDog.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_ImageNetDog.sh -------------------------------------------------------------------------------- /script/train_amazonphoto_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_amazonphoto_lp.sh -------------------------------------------------------------------------------- /script/train_amazonphoto_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_amazonphoto_nc.sh -------------------------------------------------------------------------------- /script/train_blogcatalog_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_blogcatalog_lp.sh -------------------------------------------------------------------------------- /script/train_blogcatalog_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_blogcatalog_nc.sh -------------------------------------------------------------------------------- /script/train_citeseer_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_citeseer_lp.sh -------------------------------------------------------------------------------- /script/train_citeseer_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_citeseer_nc.sh -------------------------------------------------------------------------------- /script/train_cora_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_cora_lp.sh -------------------------------------------------------------------------------- /script/train_cora_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_cora_nc.sh -------------------------------------------------------------------------------- /script/train_pubmed_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_pubmed_lp.sh -------------------------------------------------------------------------------- /script/train_pubmed_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_pubmed_nc.sh -------------------------------------------------------------------------------- /script/train_wiki_lp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_wiki_lp.sh -------------------------------------------------------------------------------- /script/train_wiki_nc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/script/train_wiki_nc.sh -------------------------------------------------------------------------------- /solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/solver.py -------------------------------------------------------------------------------- /train_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/train_solver.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/utils/math_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/junhocho/HGCAE/HEAD/utils/train_utils.py --------------------------------------------------------------------------------