gigl.src.mocking.lib.user_defined_edge_sampling

Functions

sample_hydrate_user_def_edge(mocked_dataset_info, ...)

Samples all available pos/neg edges and hydrated these edges with their features.

Module Contents

gigl.src.mocking.lib.user_defined_edge_sampling.sample_hydrate_user_def_edge(mocked_dataset_info, edge_usage_type)

Samples all available pos/neg edges and hydrated these edges with their features. e.g. for positive edge the output will be {pos_edge_src: [pos_edge_dst, [f0, f1, …, fn]]}

Parameters:

• mocked_dataset_info (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo)

• edge_usage_type (gigl.src.common.types.graph_data.EdgeUsageType)

Return type:

Dict[gigl.src.common.types.graph_data.NodeId, List]