gigl.src.common.translators.training_samples_protos_translator#

Attributes#

logger

Classes#

`NodeAnchorBasedLinkPredictionSample`
`RootedNodeNeighborhoodSample`
`SupervisedNodeClassificationSample`
`TrainingSamplesProtosTranslator`

Module Contents#

class gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample[source]#

class SampleSupervisionEdgeData[source]#

hard_neg_edge_features: torch.FloatTensor | None[source]#

hard_neg_nodes: List[gigl.src.common.types.graph_data.NodeId][source]#

pos_edge_features: torch.FloatTensor | None[source]#

pos_nodes: List[gigl.src.common.types.graph_data.NodeId][source]#

condensed_edge_type_to_supervision_edge_data: Dict[gigl.src.common.types.graph_data.CondensedEdgeType, NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData][source]#

root_node: gigl.src.common.types.graph_data.Node[source]#

subgraph: gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol[source]#

class gigl.src.common.translators.training_samples_protos_translator.RootedNodeNeighborhoodSample[source]#

Bases: NamedTuple

root_node: gigl.src.common.types.graph_data.Node[source]#

subgraph: gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol[source]#

class gigl.src.common.translators.training_samples_protos_translator.SupervisedNodeClassificationSample[source]#

Bases: NamedTuple

root_node: gigl.src.common.types.graph_data.Node[source]#

x: gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol[source]#

y: List[snapchat.research.gbml.training_samples_schema_pb2.Label][source]#

class gigl.src.common.translators.training_samples_protos_translator.TrainingSamplesProtosTranslator[source]#

static training_samples_from_NodeAnchorBasedLinkPredictionSamplePb(samples, graph_metadata_pb_wrapper, preprocessed_metadata_pb_wrapper, builder)[source]#

Parameters:

samples (List[snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample])
graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)
preprocessed_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper)
builder (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder)

Return type:

List[NodeAnchorBasedLinkPredictionSample]

static training_samples_from_RootedNodeNeighborhoodPb(samples, graph_metadata_pb_wrapper, builder)[source]#

Parameters:

samples (List[snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood])
graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)
builder (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder)

Return type:

List[RootedNodeNeighborhoodSample]

static training_samples_from_SupervisedNodeClassificationSamplePb(samples, graph_metadata_pb_wrapper, builder)[source]#

Parameters:

samples (List[snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample])
graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper)
builder (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder)

Return type:

List[SupervisedNodeClassificationSample]

gigl.src.common.translators.training_samples_protos_translator.logger[source]#