GATNE-T

GATNE [KDD2019]

Paper: [GATNE] Representation Learning for Attributed Multiplex Heterogeneous Network

Code from author: https://github.com/THUDM/GATNE

Code from dgl: https://github.com/dmlc/dgl/tree/master/examples/pytorch/GATNE-T

How to run

Clone the OpenHGNN

python main.py -m GATNE-T -t link_prediction -d HGBl-amazon -g 0 --use_best_config

If you do not have gpu, set -gpu -1.

Candidate dataset:

HGBl-amazon

Performance

Link Prediction

Evaluation metric: roc_auc

HGBl-amazon
0.7300(product-product-0) 0.8866(product-product-1) 0.8083(avgerage)

Contirbutor

Hongyi Zhang[GAMMA LAB]

If you have any questions,

Submit an issue or email to zhanghongyi@bupt.edu.cn.