graphorge.gnn_base_model.data.graph_dataset.get_subset_indices_mapping¶
- get_subset_indices_mapping(subset)[source]¶
Get mapping from subset indexes to whole parent data set indices.
- Parameters:
subset (torch.utils.data.Subset) – Subset of data set.
- Returns:
indices_map – Mapping between each subset index (key, str[int]) and the corresponding whole data set index (item, str[int]).
- Return type: