If I want to add features for my edges, can I simply replace the encoder network with a network that supports edge_attr?

PetarV- / DGI

Deep Graph Infomax (https://arxiv.org/abs/1809.10341)

MIT License

630 stars 135 forks source link

If I want to add features for my edges, can I simply replace the encoder network with a network that supports edge_attr? #18

Open ezio0218 opened 3 years ago

ezio0218 commented 3 years ago

Hi Petar, I have read your idea abot Deep Graph Infomax. While right now something really puzzles me: if I want to add features for my edges and I still want to do unsupervised or self-supervised learning. Can I just simply change the encoder and remain all the other function unchanged? Such as corruption function or readout function?