torch_geometric.data

Data Objects

Data

A data object describing a homogeneous graph.

HeteroData

A data object describing a heterogeneous graph, holding multiple node and/or edge types in disjunct storage objects.

Batch

A data object describing a batch of graphs as one big (disconnected) graph.

TemporalData

A data object composed by a stream of events describing a temporal graph.

Dataset

Dataset base class for creating graph datasets.

InMemoryDataset

Dataset base class for creating graph datasets which easily fit into CPU memory.

Remote Backend Interfaces

FeatureStore

An abstract base class to access features from a remote feature store.

GraphStore

An abstract base class to access edges from a remote graph store.

TensorAttr

Defines the attributes of a FeatureStore tensor.

EdgeAttr

Defines the attributes of a GraphStore edge.

PyTorch Lightning Wrappers

LightningDataset

Converts a set of Dataset objects into a pytorch_lightning.LightningDataModule variant.

LightningNodeData

Converts a Data or HeteroData object into a pytorch_lightning.LightningDataModule variant.

LightningLinkData

Converts a Data or HeteroData object into a pytorch_lightning.LightningDataModule variant.

Helper Functions

makedirs

Recursively creates a directory.

download_url

Downloads the content of an URL to a specific folder.

extract_tar

Extracts a tar archive to a specific folder.

extract_zip

Extracts a zip archive to a specific folder.

extract_bz2

Extracts a bz2 archive to a specific folder.

extract_gz

Extracts a gz archive to a specific folder.