pytorch-geometric

Split DBP15K from pytorch geometric in train test and validation

Split DBP15K from pytorch geometric in train test and validation Question: I have a code in which I use the DBP15K dataset via from torch_geometric.datasets import DBP15K data = DBP15K(path, args.category, transform=SumEmbedding())[0].to(device) But according to the documentation of pytorch geometric this one is divided only in train and in test. I tried to divide it …

Total answers: 1

Pytorch Geometric: 'from_networkx' doesn't work with my 'group_node_attrs'

Pytorch Geometric: 'from_networkx' doesn't work with my 'group_node_attrs' Question: I’ trying to convert a NetwrokX graph into the pyg format to feed it to a GCN. from_networkx(G) works without problems from_networkx(G, group_node_attrs=x) # doesn’t work, and I get the following error: Here the documentation about how the function ‘from_networkx’: https://pytorch-geometric.readthedocs.io/en/latest/_modules/torch_geometric/utils/convert.html Traceback (most recent call last): …

Total answers: 1

Visualising the last layer node embeddings of a model in torch geometric

Visualising the last layer node embeddings of a model in torch geometric Question: I’m doing my first graph convolutional neural network project with torch_geometric. I want to visualize the last layer node embeddings of my model and don’t know how I should get it. I trained my model on the CiteSeer dataset. You can get …

Total answers: 1