/Metropolis-Hastings-Data-Augmentation-for-Graph-Neural-Networks

Official PyTorch Implementation of "Metropolis-Hastings Data Augmentation for Graph Neural Networks". NeurIPS 2021.

Primary LanguagePython

Metropolis-Hastings Data Augmentation for Graph Neural Networks

This repository is the implementation of MH-Aug.

Hyeonjin Park* , Seunghun Lee* , Sihyeon Kim, Jinyoung Park, Jisu Jeong, Kyung-min Kim, Jung-Woo Ha, Hyunwoo J. Kim, In Advanced in Neural Information Processing Systems (NeurIPS 2021).

Required packages

A list of dependencies will need to be installed in order to run the code. We provide the dependency yaml file (env.yml)

$ conda env create -f env.yml

Running the code

$ python main.py

Citation

@article{park2021metropolis,
  title={Metropolis-Hastings Data Augmentation for Graph Neural Networks},
  author={Park, Hyeonjin and Lee, Seunghun and Kim, Sihyeon and Park, Jinyoung and Jeong, Jisu and Kim, Kyung-Min and Ha, Jung-Woo and Kim, Hyunwoo J},
  journal={Advances in Neural Information Processing Systems},
  volume={34},
  year={2021}
}

License

Copyright (c) 2021-present NAVER Corp. and Korea University