Skip to content

Refactor Embedding and GNN

Anthony Correia requested to merge anthonyc/refactor_embedding_gnn into main

To merge after !50

For consistency:

  • Try to improve how the forward pass of the GNN is written
  • Rename edge_indices to edge_index
  • Rename dict_triplet_indices to triplet_indices and triplet_indices to triplet_index
  • Rename edge_output and triplet_outputs to edge_logit and triplet_logits (respectively)
Edited by Anthony Correia

Merge request reports