Theheavens d316cd780c | 2 years ago | |
---|---|---|
.. | ||
README.md | 2 years ago |
Paper:Unsupervised Attributed Multiplex Network Embedding
Code from author:https://github.com/pcy1302/DMGI
Clone the Openhgnn-DGL
python main.py -m DMGI -t node_classification -d acm_han_raw -g 0 --use_best_config
Candidate dataset: acm_han_raw
If you do not have gpu, set -gpu -1.
acm_han_raw/imdb4GTN
NOTE: DMGI can handle imdb dataset, we will add the dataset in our further work.
Node classification
Node classification | acm | imdb4GTN |
---|---|---|
paper | 89.8 | --- |
OpenHGNN | 89.73 | 52.52 |
The model is trained in unsupervisied node classification.
learning_rate = 0.0005
l2_coef = 0.0001
sc = 3
dropout = 0.5
reg_coef = 0.001
sup_coef = 0.1
patience =20
hid_unit = 64
num_heads = 1
max_epoch = 10000
isSemi = False
isBias = False
isAttn = False
Best config can be found in best_config
Siyuan Zhang, Tianyu Zhao[GAMMA LAB]
Submit an issue or email to 1580124318@qq.com.
OpenHGNN是由北邮GAMMA Lab开发的基于PyTorch和DGL的开源异质图神经网络工具包。
Python Markdown Shell
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》