gigl.src.mocking.dataset_asset_mocking_suite#

Attributes#

`logger`
`parser`

Classes#

DatasetAssetMockingSuite

This class houses functions which are used to mock datasets for testing purposes,

Module Contents#

class gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite[source]#

This class houses functions which are used to mock datasets for testing purposes, e.g. mock_cora_homogeneous_supervised_node_classification_dataset. To add a mocking task, create a new function which starts with mock and returns a MockedDatasetInfo instance.

class ToyGraphData[source]#

edge_feats: Dict[str, torch.Tensor] | None = None[source]#

edge_indices: Dict[str, torch.Tensor][source]#

edge_types: Dict[str, gigl.src.common.types.graph_data.EdgeType][source]#

node_feats: Dict[str, torch.Tensor][source]#

node_labels: Dict[str, torch.Tensor] | None = None[source]#

node_types: Dict[str, gigl.src.common.types.graph_data.NodeType][source]#

class UserDefinedLabels[source]#

neg_edge_feats: torch.Tensor[source]#

neg_edge_index: torch.Tensor[source]#

pos_edge_feats: torch.Tensor[source]#

pos_edge_index: torch.Tensor[source]#

compute_datasets_to_mock(selected_datasets=None)[source]#

Returns a dictionary of mocked datasets to be used in the mocking suite. If selected_datasets is provided, only those datasets will be returned.

Parameters:: selected_datasets (Optional[List[str]])
Return type:: Dict[str, gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo]

mock_cora_homogeneous_node_anchor_based_link_prediction_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_cora_homogeneous_node_anchor_based_link_prediction_dataset_with_edge_features()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_cora_homogeneous_node_anchor_based_link_prediction_dataset_with_user_defined_labels()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_cora_homogeneous_supervised_node_classification_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_cora_homogeneous_supervised_node_classification_dataset_with_edge_features()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_dblp_node_anchor_based_link_prediction_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_dblp_node_anchor_based_link_prediction_dataset_with_user_defined_labels()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_toy_graph_heterogeneous_node_anchor_based_link_prediction_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_toy_graph_homogeneous_node_anchor_based_link_prediction_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

mock_toy_graph_homogeneous_node_anchor_based_link_prediction_with_user_def_labels_dataset()[source]#

Return type:: gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo

gigl.src.mocking.dataset_asset_mocking_suite.logger[source]#

gigl.src.mocking.dataset_asset_mocking_suite.parser[source]#