gigl.src.common.translators.NodeAnchorBasedLinkPredictionSample#
- class gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample(root_node: gigl.src.common.types.graph_data.Node, subgraph: gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol, condensed_edge_type_to_supervision_edge_data: Dict[CondensedEdgeType, gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData])#
Bases:
objectMethods
- class SampleSupervisionEdgeData(pos_nodes: List[NodeId], hard_neg_nodes: List[NodeId], pos_edge_features: torch.FloatTensor | None, hard_neg_edge_features: torch.FloatTensor | None)#
Bases:
object- __eq__(other)#
Return self==value.
- __hash__ = None#
- __init__(pos_nodes: List[NodeId], hard_neg_nodes: List[NodeId], pos_edge_features: FloatTensor | None, hard_neg_edge_features: FloatTensor | None) None#
- __repr__()#
Return repr(self).
- __weakref__#
list of weak references to the object (if defined)
- __eq__(other)#
Return self==value.
- __hash__ = None#
- __init__(root_node: Node, subgraph: GbmlGraphDataProtocol, condensed_edge_type_to_supervision_edge_data: Dict[CondensedEdgeType, SampleSupervisionEdgeData]) None#
- __repr__()#
Return repr(self).
- __weakref__#
list of weak references to the object (if defined)