Skip to content
/ RHINE Public

Source code for AAAI 2019 paper "Relation Structure-Aware Heterogeneous Information Network Embedding"

Notifications You must be signed in to change notification settings

rootlu/RHINE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

7 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

RHINE

Source code for AAAI 2019 paper "Relation Structure-Aware Heterogeneous Information Network Embedding"

Requirements

  • Python 2.7

  • numpy

  • scipy

  • PyTorch (0.3.0)

  • My machine with two GPUs (NVIDIA GTX-1080 *2) and two CPUs (Intel Xeon E5-2690 * 2)

Description

RHINE/
β”œβ”€β”€ code
β”‚   β”œβ”€β”€ config
β”‚   β”‚   β”œβ”€β”€ Config.py:configs for model.
β”‚   β”‚   └──_init_.py
β”‚   β”œβ”€β”€ evaluation.py: evaluate the performance of learned embeddings w.r.t clustering and classification
β”‚   β”œβ”€β”€ models
β”‚   β”‚   β”œβ”€β”€ _init_.py
β”‚   β”‚   β”œβ”€β”€ Model.py: the super model with some functions
β”‚   β”‚   └── RHINE.py: our model
β”‚   β”œβ”€β”€ preData
β”‚   β”‚   └── dblpDataHelper.py: data preparation for our mode
β”‚   β”œβ”€β”€ release
β”‚   β”‚   β”œβ”€β”€ Sample_ARs.so: sampling with dll
β”‚   β”‚   └── Sample_IRs.so
β”‚   └── trainRHINE.py: train model
β”œβ”€β”€ data
β”‚   └── dblp
β”‚       β”œβ”€β”€ node2id.txt: the first line is the number of nodes, (node_type+node_name, node_id)
β”‚       β”œβ”€β”€ paper_label.txt: (node_name, label)
β”‚       β”œβ”€β”€ relation2id.txt: the first line is the number of relations, 		   (relation_name, relation_id)
β”‚       β”œβ”€β”€ train2id_apc.txt:  (node1_id, node2_id, relation_id, weight)
β”‚       β”œβ”€β”€ train2id_pc.txt
β”‚       β”œβ”€β”€ train2id_ap.txt
β”‚       β”œβ”€β”€ train2id_pt.txt
β”‚       β”œβ”€β”€ train2id_apt.txt
β”‚       β”œβ”€β”€ train2id_ARs.txt: the first line is the number of ARs triples, (node1_id, node2_id, relation_id, weight)
β”‚       └── train2id_IRs.txt
β”œβ”€β”€ README.md
└── res
    └── dblp
        └── embedding.vec.ap_pt_apt+pc_apc.json: the learned embeddings 

Reference

@inproceedings{Yuanfu2019RHINE,
  title={Relation Structure-Aware Heterogeneous Information Network Embedding},
  author={Yuanfu Lu, Chuan Shi, Linmei Hu, Zhiyuan Liu.}
  booktitle={Proceedings of AAAI},
  year={2019}
}

About

Source code for AAAI 2019 paper "Relation Structure-Aware Heterogeneous Information Network Embedding"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages