GNNs Prof. Anderson Rocha Tavares Slide 23 / 24

Split Correto em Link Prediction

Bloco 4 — Treino com poucas arestas, expandindo em val/test

Aresta de mensagem (linha contínua; entra no GNN) Aresta de supervisão (linha tracejada; alvo da etapa)
1) Grafo completo (referência)
A B C D E F
Todas as conexões disponíveis no dataset original.
2) Treino (menos arestas)
A B C D E F
Treino começa com o menor grafo de mensagem.
3) Validação
A B C D E F
Arestas-alvo de val continuam fora da mensagem nessa etapa.
4) Teste
A B C D E F
No teste, apenas novas arestas-alvo ficam fora do message passing.
Regra: na etapa atual, arestas de supervisão nunca aparecem como arestas de mensagem.