gigl.common.data.SerializedTFRecordInfo#

class gigl.common.data.dataloaders.SerializedTFRecordInfo(tfrecord_uri_prefix: Uri, feature_keys: Sequence[str], feature_spec: Dict[str, FixedLenFeature | VarLenFeature | SparseFeature | RaggedFeature], feature_dim: int, entity_key: str | Tuple[str, str], tfrecord_uri_pattern: str = '.*tfrecord(.gz)?$')#

Bases: object

Stores information pertaining to how a single entity (node, edge, positive label, negative label) and single node/edge type in the heterogeneous case is serialized on disk. This field is used as input to the TFRecordDataLoader.load_as_torch_tensor() function for loading torch tensors.

Methods

__init__

__delattr__(name)#: Implement delattr(self, name).

__eq__(other)#: Return self==value.

__hash__()#: Return hash(self).

__init__(tfrecord_uri_prefix: Uri, feature_keys: Sequence[str], feature_spec: Dict[str, FixedLenFeature | VarLenFeature | SparseFeature | RaggedFeature], feature_dim: int, entity_key: str | Tuple[str, str], tfrecord_uri_pattern: str = '.*tfrecord(.gz)?$') → None#

__repr__()#: Return repr(self).

__setattr__(name, value)#: Implement setattr(self, name, value).

__weakref__#: list of weak references to the object (if defined)

property is_node_entity: bool#: Returns whether this serialized entity contains node or edge information by checking the type of entity_key