How to add nodes to graphs in loaded dataset

[simon-forb]

What would be the most canonical way to add nodes to a dataset that is imported through torch_geometric?

For instance, I am loading the reddit binary dataset like this:

reddit = torch_geometric.datasets.TUDataset("/tmp/reddit-b", name="REDDIT-BINARY")

This yields a dataset which contains 2000 graphs. Now I would like to add a number of nodes to each graph. I only figured out how to do this using networkx but I believe there is probably a better way to do this without this double conversion.

[rusty1s]

You can modify graphs in Dataset through the transform/pre_transform interface. For how to add new nodes, take a look at, e.g., the VirtualNode transform. In general, it is as simple as

data.edge_index = torch.cat([data.edge_index, torch.tensor(new_edges)], dim=1)
data.x = torch.cat([data.x, new_features], dim=0)