gigl.src.mocking.lib.mock_input_for_trainer#

Attributes#

logger

Functions#

`split_and_write_node_anchor_link_prediction_subgraph_samples_from_mocked_dataset_info`(...)
`split_and_write_supervised_node_classification_subgraph_samples_from_mocked_dataset_info`(...)

Module Contents#

gigl.src.mocking.lib.mock_input_for_trainer.split_and_write_node_anchor_link_prediction_subgraph_samples_from_mocked_dataset_info(mocked_dataset_info, sample_edge_type, gbml_config_pb, hetero_data)[source]#

Parameters:

mocked_dataset_info (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo)
sample_edge_type (gigl.src.common.types.graph_data.EdgeType)
gbml_config_pb (snapchat.research.gbml.gbml_config_pb2.GbmlConfig)
hetero_data (torch_geometric.data.HeteroData)

gigl.src.mocking.lib.mock_input_for_trainer.split_and_write_supervised_node_classification_subgraph_samples_from_mocked_dataset_info(mocked_dataset_info, root_node_type, gbml_config_pb, hetero_data)[source]#

Parameters:

mocked_dataset_info (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo)
root_node_type (gigl.src.common.types.graph_data.NodeType)
gbml_config_pb (snapchat.research.gbml.gbml_config_pb2.GbmlConfig)
hetero_data (torch_geometric.data.HeteroData)

gigl.src.mocking.lib.mock_input_for_trainer.logger[source]#