torch_geometric.data
Data Objects
A data object describing a homogeneous graph. |
|
A data object describing a heterogeneous graph, holding multiple node and/or edge types in disjunct storage objects. |
|
A data object describing a batch of graphs as one big (disconnected) graph. |
|
A data object composed by a stream of events describing a temporal graph. |
|
Dataset base class for creating graph datasets. |
|
Dataset base class for creating graph datasets which easily fit into CPU memory. |
|
Dataset base class for creating large graph datasets which do not easily fit into CPU memory at once by leveraging a |
|
For a dataset that consists of multiple subgraphs that are assumed to be part of a much larger graph, collate the values into a large graph store to save resources. |
|
Remote Backend Interfaces
An abstract base class to access features from a remote feature store. |
|
An abstract base class to access edges from a remote graph store. |
|
Defines the attributes of a |
|
Defines the attributes of a |
Databases
Base class for inserting and retrieving data from a database. |
|
An index-based key/value database based on |
|
An index-based key/value database based on |
PyTorch Lightning Wrappers
Converts a set of |
|
Converts a |
|
Converts a |
Helper Functions
Recursively creates a directory. |
|
Downloads the content of an URL to a specific folder. |
|
Downloads the content of a Google Drive ID to a specific folder. |
|
Extracts a tar archive to a specific folder. |
|
Extracts a zip archive to a specific folder. |
|
Extracts a bz2 archive to a specific folder. |
|
Extracts a gz archive to a specific folder. |
|
For a given set of triplets retrieve a Data object containing the |
|
Given an indexer and a series of triplet groups (like a dataset), retrieve the specified node and edge features for each triplet from the index. |