search

zhao-tong / GAug

AAAI'21: Data Augmentation for Graph Neural Networks
MIT License
186 stars 34 forks source link

File function #1

Closed wangzeyu135798 closed 3 years ago

wangzeyu135798 commented 4 years ago

Hi: I read your paper recently. I thought it great. I have two questions.

  1. What are the function of four optuna_.py files?
  2. How to make new edge_probabilities files by GAE mentioned ih paper? There are only two files provied in folder. Thanks.
JunweiSUN commented 3 years ago

Same questions here. @GAugAuthors Can you guys provide the code of generating M? Should I use a well trained GAE or just use a 2-layer GCN and do the forward process?

zhao-tong commented 3 years ago

Sry for the late reply.

  1. The optuna_.py files are for hyperparameter search with the optuna package.
  2. I used VGAE (https://github.com/DaehanKim/vgae_pytorch) for generating the edge_probabilities for GAugM.
zhao-tong commented 3 years ago

I just added the VGAE implementation I used in this repo. Closing this issue.