gigl.src.common.models.layers.task

Classes

BGRL	Creates 2 augmented views of input graph with augmentations 1 and 2 and defines task-specific linear head and BGRL loss
DirectAU	DirectAU (https://arxiv.org/pdf/2206.12811.pdf) optimizes for representation quality in Collaborative Filtering from the perspective of alignment and uniformity on the hypersphere.
FeatureReconstruction	Masks out percentage of anchor nodes' features before attempting to recreate these embeddings
GBT	Creates 2 augmented views of input graph with augmentations 1 and 2 and defines task-specific linear head for GBT loss
GRACE	Creates 2 augmented views of input graph with augmentations 1 and 2 and defines task-specific linear head for GRACE Loss
Margin
NodeAnchorBasedLinkPredictionBaseTask
NodeAnchorBasedLinkPredictionTasks
Retrieval
Softmax
TBGRL	Creates 3 augmented views of input graph with positive augmentations 1, 2 and negative augmentation 3 and defines task-specific linear head and TBGRL loss
WhiteningDecorrelation	Creates 2 augmented views of input graph with augmentations 1 and 2 and defines task-specific linear head for Whitening Decorrelation Loss