PyTorch implementation for Graph Contrastive Learning Automated [talk] [poster] [appendix]
Yuning You, Tianlong Chen, Yang Shen, Zhangyang Wang
In ICML 2021.
In this repository, we propose a principled framework named joint augmentation selection (JOAO), to automatically, adaptively and dynamically select augmentations during GraphCL training. Sanity check shows that the selection aligns with previous “best practices”, as shown in Figure 2.
If you use this code for you research, please cite our paper.
@article{you2021graph,
title={Graph Contrastive Learning Automated},
author={You, Yuning and Chen, Tianlong and Shen, Yang and Wang, Zhangyang},
journal={arXiv preprint arXiv:2106.07594},
year={2021}
}
Author: Shen-Lab
Download Link: Download The Source Code
Official Website: https://github.com/Shen-Lab/GraphCL_Automated
#pytorch #python #machine-learning