Link Prediction and Graph Classification
Edge prediction task, negative sampling, graph-level pooling, GINConv for graph classification.
Two New Graph Tasks
Beyond classifying nodes, GNNs handle:
- Link prediction: will an edge exist between two nodes? (friend suggestions, drug interactions)
- Graph classification: assign a label to an entire graph (is this molecule toxic?)
Link Prediction Setup
In link prediction we first compute node embeddings with a GNN, then score candidate node pairs. A high score means the model believes an edge should connect them.
All lessons in this course
- Graph Theory for Machine Learning
- Graph Convolutional Networks (GCN)
- Node Classification with GNN
- Link Prediction and Graph Classification