使用pyg的时候用Karateclub数据集出现上述错误,出错代码如下
from torch_geometric.datasets import LRGBDataset, KarateClub
from torch_geometric.utils import to_networkx
data = KarateClub()
G = to_networkx(data)
发现如果直接使用data放到to_networkx函数里会报错
使用print(data)发现输出的结果是
KarateClub()
使用print(data[0])发现输出的结果是
Data(x=[34, 34], edge_index=[2, 156], y=[34], train_mask=[34])
问题很明显了,应该用G = to_networkx(data[0])