InfoGraph
InfoGraph copied to clipboard
Official code for "InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization" (ICLR 2020, spotlight)
When I look at : ``` error += (model(data) * std - data.y * std).abs().sum().item() # MAE ``` I am curious what the .sum is doing here? Aren't these just...
Hi, I find that as the training goes (beyond 20 epochs), the loss will gradually become negative. May I ask if this is harmful to downstream tasks? Thank you!
Hi, I am trying to run your framework on my own unlabeled dataset. However, as I looked into your code of the **unsupervised** section, I see in your `get_embedding()` function...