Skip to content

Latest commit

 

History

History

baseline

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
utils
utils
 
 
GNN.py
GNN.py
 
 
README.md
README.md
 
 
conv.py
conv.py
 
 
run.py
run.py
 
 
run_multi.py
run_multi.py
 
 
run_new.py
run_new.py
 
 

README.md

Simple-HGN for HGB

For message passing with relation attention version:

python run_new.py --dataset DBLP
python run_new.py --dataset ACM --feats-type 2
python run_multi.py --dataset IMDB --feats-type 0
python run_new.py --dataset Freebase

running environment

  • torch 1.6.0 cuda 10.1
  • dgl 0.4.3 cuda 10.1
  • networkx 2.3
  • scikit-learn 0.23.2
  • scipy 1.5.2