yeze16159 / CurvGN

Curvature Graph Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CurvGN

Curvature Graph Network

Requirement:

  • numpy

  • torch

  • torch_geometric

  • networkx

  • GraphRicciCurvature

Run BenchMark:

python pipelines.py

Run synthetic experiment:

cd syndata
python Ranset_gen.py
cd ..
./pipelines_heat.sh

Cite:

Please cite our paper in following if you use this code in your work:

@inproceedings{ze20curvature,
   title="Curvature Graph Network",
   author="Ze Ye and Kin Sum Liu and Tengfei Ma and Jie Gao and Chao Chen",
   booktitle="Proceedings of the 8th International Conference on Learning Representations (ICLR 2020)",
   month="April",
   year="2020"
}

About

Curvature Graph Network


Languages

Language:Python 95.3%Language:Shell 4.7%