Skip to content

Latest commit

 

History

History

dhn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

Distance encoding based Heterogeneous graph neural Network (DHN)

Dataset Statics

Dataset # Nodes # Edges
acm 3908 4500

Results

TL_BACKEND="torch" python dhn_trainer.py --test_ratio 0.3 --one_hot True --k_hop 2 --num_neighbor 5 --batch_size 32 --lr 0.001 --n_epoch 100 --drop_rate 0.01 --dataset 'acm'
Dataset Paper(AUC) Our(th)(AUC)
acm 95.07 95.54±0.18