PyTorch Geometric GCN 图神经网络实现：使用自定义数据集进行训练和测试

在上述代码中，edges.txt边连接关系文件中，第1列为源节点，第2列为目标节点。根据这个关系，在read_edges()函数中，需要将读取的边连接关系数据进行转置操作，即将t()去掉。修改后的代码如下所示：\n\npython\ndef read_edges(self):\n edges_path = self.root + '/edges.txt'\n with open(edges_path, 'r') as f:\n edges = [list(map(int, line.split())) for line in f.readlines()]\n return torch.tensor(edges, dtype=torch.long)\n\n\n这样修改后，每一行的第一列就是源节点，第二列就是目标节点。