Skip to content

ronghangzhu/PBT-STN-GCN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 

Repository files navigation

PBT-STN-GCN

This is the code for paper "Automated Graph Learning via Population based Self-Tuning GCN". It only includes STN-GCN code, we will release the PBT-STN-GCN code later.

Prerequisites

  • Python3
  • Pytorch == 1.0.0 (with suitable CUDA and CuDNN version)

Dataset

The datasets (Cora, Citeseer and PubMed) are in GoogleDrive and BaiduPan (pw:frvg).
You need to create a "./data" file in "./citeseer" file and move the dataset into it.

Training

You can run python train.py to train and evaluate.

Citation

If you use this code for your research, please consider citing:

@inproceedings{zhu2021automated,
  title={Automated Graph Learning via Population Based Self-Tuning GCN},
  author={Zhu, Ronghang and Tao, Zhiqiang and Li, Yaliang and Li, Sheng},
  booktitle={Proceedings of the 44th International ACM SIGIR Conference on Research and Development in Information Retrieval},
  pages={2096--2100},
  year={2021}
}

About

This is the code for paper "Automated Graph Learning via Population based Self-Tuning GCN"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages