Contents
PyTorch Geometric (PyG) is a geometric deep learning extension library for PyTorch.
torch_geometric.data
共以下十个类:
- 单(个/批)图数据:
- Data: A plain old python object modeling a single graph with various (optional) attributes
- Batch: A plain old python object modeling a batch of graphs as one big (dicconnected) graph.
- With
torch_geometric.data.Data
being the base class, all its methods can also be used here. - In addition, single graphs can be reconstructed via the assignment vector
batch
, which maps each node to its respective graph identifier.
- With
- 为数据集创建提供的两个抽象类( 官方教程 ):
- Dataset: Dataset base class for creating graph datasets.
- InMemoryDataset: … fit completely into memory. ( torch_geometric.datasets 中的现有数据集多从此类继承)