torch_geometric.transforms.BaseTransform
- class BaseTransform[source]
Bases:
ABC
An abstract base class for writing transforms.
Transforms are a general way to modify and customize
Data
orHeteroData
objects, either by implicitly passing them as an argument to aDataset
, or by applying them explicitly to individualData
orHeteroData
objects:import torch_geometric.transforms as T from torch_geometric.datasets import TUDataset transform = T.Compose([T.ToUndirected(), T.AddSelfLoops()]) dataset = TUDataset(path, name='MUTAG', transform=transform) data = dataset[0] # Implicitly transform data on every access. data = TUDataset(path, name='MUTAG')[0] data = transform(data) # Explicitly transform data.