This is an open-source toolkit for Heterogeneous Information Network Embedding(OpenHINE) with version 0.1. We can train and test the model more easily. It provides implementations of many popular models, including: DHNE, HAN, HeGAN, HERec, HIN2vec, Metapath2vec, MetaGraph2vec, RHINE. More materials can be found in www.shichuan.org.

convenience provided:

• ​ easy to train and evaluate
• ​ able to extend new/your datasets and models
• ​ the latest model available: HAN、HeGAN and so on

DMGroup from BUPT: Tianyu Zhao, Meiqi Zhu, Jiawei Liu, Nian Liu, Guanyi Chu, Jiayue Liu, Xiao Wang, Cheng Yang, Linmei Hu, Chuan Shi.

• Python version >= 3.6

• PyTorch version >= 1.4.0

• TensorFlow version >= 1.14

• Keras version >= 2.3.1

python train.py -m model_name -d dataset_name

e.g.

python train.py -m Metapath2vec -d acm

The model parameter could be modified in the file ( ./src/config.ini ).

• common parameter

​ --alpha: learning rate

​ --dim: dimension of output

​ --epoch: the number of iterations

​ etc...

• specific parameter

​ --metapath: the metapath selected

​ --neg_num: the number of negative samples

​ etc...

If you want to train your own dataset, create the file (./dataset/your_dataset_name/edge.txt) and the format is as follows：

​ src_node_id dst_node_id edge_type weight

​ e.g.

	19	7	p-c	2
	19	7	p-a	1
	11	0	p-c	1
	0	11	c-p	1

PS：The input graph is directed and the undirected needs to be transformed into directed graph.

​ number_of_nodes embedding_dim

​ node_name dim1 dim2

e.g.

11246	2
a1814 0.06386946886777878 -0.04781734198331833
a0 ... ...

​ Structural Deep Embedding for Hyper-Networks

​ src code:https://github.com/tadpole/DHNE

​ Heterogeneous Graph Attention Network

​ Add feature.txt into the input folder or set the parameter "featype": "adj", which means that you will use adjacency matrix as your feature.

​ src code:https://github.com/Jhy1993/HAN

​ Adversarial Learning on Heterogeneous Information Network

​ src code:https://github.com/librahu/HeGAN

​ Heterogeneous Information Network Embedding for Recommendation

​ src code:https://github.com/librahu/HERec

​ metapath_list: pap|psp (split by "|")

​ HIN2Vec: Explore Meta-paths in Heterogeneous Information Networks for Representation Learning

​ src code:https://github.com/csiesheep/hin2vec

​ metapath2vec: Scalable Representation Learning for Heterogeneous Networks

​ src code:https://ericdongyx.github.io/metapath2vec/m2v.html

​ the python version implemented by DGL:https://github.com/dmlc/dgl/tree/master/examples/pytorch/metapath2vec

​ MetaGraph2Vec: Complex Semantic Path Augmented Heterogeneous Network Embedding

​ src code:https://github.com/daokunzhang/MetaGraph2Vec

​ Relation Structure-Aware Heterogeneous Information Network Embedding

​ only supported in the Linux

​ src code:https://github.com/rootlu/RHINE 

python test.py -d dataset_name -m model_name -n file_name

The output embedding file name can be found in (./output/embedding/model_name/) .

e.g.

python test.py -d dblp -m HAN -n node.txt

​ number_of_nodes embedding_dim

​ node_name dim1 dim2

e.g.

11246	2
a1814 0.06386946886777878 -0.04781734198331833
a0 ... ...

HAN uses the dataset without features.

Note that OpenHINE is just version 0.1 and still actively under development, so feedback and contributions are welcome. Feel free to submit your questions as a issue.

In the future, we will contain more models and tasks. We use the assorted deep learning framework, so we want to unify the model with PyTorch. If you have a demo of the above model with PyTorch or want your method added into our toolkit, contract us please.