GraphCL
GraphCL copied to clipboard
[NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen
File "/home/zhenghua/pythoncode/unsupervised_graph_TU/gsimclr.py", line 166, in dataset = TUDataset(path, name=DS, aug=args.aug).shuffle() File "/home/zhenghua/pythoncode/unsupervised_graph_TU/aug.py", line 73, in __init__ pre_filter) File "/home/zhenghua/.conda/envs/graphcontra/lib/python3.6/site-packages/torch_geometric/data/in_memory_dataset.py", line 56, in __init__ pre_filter) File "/home/zhenghua/.conda/envs/graphcontra/lib/python3.6/site-packages/torch_geometric/data/dataset.py", line 76, in __init__...
Hi, I have read the paper and I want to know how to change the label rate of this?
600 1 ================ lr: 0.01 num_features: 1 hidden_dim: 32 num_gc_layers: 3 ================ /opt/conda/conda-bld/pytorch_1623448224956/work/aten/src/ATen/native/cuda/Indexing.cu:702: indexSelectLargeIndex: block: [158,0,0], thread: [105,0,0] Assertion `srcIndex < srcSelectDimSize` failed. /opt/conda/conda-bld/pytorch_1623448224956/work/aten/src/ATen/native/cuda/Indexing.cu:702: indexSelectLargeIndex: block: [158,0,0], thread: [55,0,0] Assertion...
I have a question about these two datasets. there are three subset trainset, valset, and testset. How to combine these dataset to one whole set? Thank you. @yongduosui
There are no the script "from train.train_SBMs_node_classification import train_epoch, evaluate_network"
When I run main_superpixels_contrastive.py I got f = pickle.load(f) ModuleNotFoundError: No module named 'data.superpixels' I don't know why "f = pickle.load(f)" can cause this problem?
pip install torch-scatter==1.1.0 -f https://pytorch-geometric.com/whl/torch-1.4.0.html pip install torch-sparse==0.4.4 -f https://pytorch-geometric.com/whl/torch-1.4.0.html pip install torch-cluster==1.4.5 -f https://pytorch-geometric.com/whl/torch-1.4.0.html pip install torch-spline-conv==1.1.0 -f https://pytorch-geometric.com/whl/torch-1.4.0.html pip install torch-vision==0.5.0 I am trying to install these but...
Thanks for your awesome codes, but I have some questions about `./unsupervised_TU/gsimclr.py` file. In the `loss_cal` function, the denominator will minus the positive pair `loss = pos_sim / (sim_matrix.sum(dim=1) -...
Hi, I have the problem of https://github.com/Shen-Lab/GraphCL/issues/4#issuecomment-742254458 and #1 trying lunching semisupervised TU pre training. I launch `python main.py --dataset MUTAG --aug1 random2 --aug2 random2 --lr 0.001 --suffix 0 --exp...
Hello, When running the code, I find there is a bug from main_superpixels_contrastive.py. It reports as follows:  It is a DGL batch graph index bug. Could you fix it?...