HongyangGao / Graph-U-Nets

Pytorch implementation of Graph U-Nets (ICML19)
http://proceedings.mlr.press/v97/gao19a/gao19a.pdf
GNU General Public License v3.0
513 stars 100 forks source link

Dropout, Normalization #12

Closed vthost closed 4 years ago

vthost commented 4 years ago

Hi, in the paper, you mention dropout values on both the node features and the adjacency matrix. Could you please point me to that part in the code? I have trouble to find it. I also wanted to use the pytorch geometric version of GraphUNet, instead of the one in ops. Do I then still have to normalize the adjacency matrix before (A = ops.normalize_adj(n2n_sp))? I think this might be covered within GCNConv in GraphUNet, but am not sure. Thank you already!