gigl.src.common.utils.gbml_protos#

Classes#

TrainingSamplesSchemaProtoUtils

Module Contents#

class gigl.src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils[source]#

static build_NodeAnchorBasedLinkPredictionSamplePb(target_node, target_neighborhood, pos_neighborhoods, hard_neg_neighborhoods=None, random_neg_neighborhoods=None)[source]#

Parameters:

target_node (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper)
target_neighborhood (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper)
pos_neighborhoods (list[Tuple[gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper, gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper]])
hard_neg_neighborhoods (Optional[list[Tuple[gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper, gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper]]])
random_neg_neighborhoods (Optional[list[Tuple[gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper, gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper]]])

Return type:

snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample

static build_RootedNodeNeighborhoodPb(root_node, neighborhood)[source]#

Parameters:

root_node (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper)
neighborhood (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper)

Return type:

snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood

static build_SupervisedLinkBasedTaskSamplePb()[source]#

Return type:: snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample

static build_SupervisedNodeClassificationSamplePb(target_node, neighborhood, node_labels)[source]#

Parameters:

target_node (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper)
neighborhood (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper)
node_labels (list[snapchat.research.gbml.training_samples_schema_pb2.Label])

Return type:

snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample