gigl.src.training.v1.lib.data_loaders.SupervisedNodeClassificationBatch

class gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch(graph: 'Union[torch_geometric.data.Data, torch_geometric.data.hetero_data.HeteroData]', root_node_indices: 'torch.LongTensor', root_nodes: 'List[Node]', root_node_labels: 'Optional[torch.LongTensor]' = None)

Bases: object

Methods

__init__
collate_pyg_node_classification_minibatch
get_default_data_loader
preprocess_node_classification_raw_sample_fn
preprocess_node_classification_sample_fn
process_raw_pyg_samples_and_collate_fn

__eq__(other)

Return self==value.

__hash__ = None

__init__(graph: Data | HeteroData, root_node_indices: LongTensor, root_nodes: List[Node], root_node_labels: LongTensor | None = None) → None

__repr__()

Return repr(self).

__weakref__

list of weak references to the object (if defined)