torch_geometric.datasets.AttributedGraphDataset
- class AttributedGraphDataset(root: str, name: str, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None)[source]
Bases:
InMemoryDataset
A variety of attributed graph datasets from the “Scaling Attributed Network Embedding to Massive Graphs” paper.
- Parameters
root (str) – Root directory where the dataset should be saved.
name (str) – The name of the dataset (
"Wiki"
,"Cora"
"CiteSeer"
,"PubMed"
,"BlogCatalog"
,"PPI"
,"Flickr"
,"Facebook"
,"Twitter"
,"TWeibo"
,"MAG"
).transform (callable, optional) – A function/transform that takes in an
torch_geometric.data.Data
object and returns a transformed version. The data object will be transformed before every access. (default:None
)pre_transform (callable, optional) – A function/transform that takes in an
torch_geometric.data.Data
object and returns a transformed version. The data object will be transformed before being saved to disk. (default:None
)