iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

comment edgelist

Browse Source
This commit is contained in:
mhz
2024-07-05 10:49:43 +02:00
parent deadd50d24
commit 9601a3c18d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
graph_dit/datasets/dataset.py
View File

@@ -692,8 +692,8 @@ class Dataset(InMemoryDataset):
if adj[start][end] == 1: if adj[start][end] == 1:
edges_list.append((start, end)) edges_list.append((start, end))
edge_type.append(1) edge_type.append(1)
edges_list.append((end, start)) # edges_list.append((end, start))
edge_type.append(1) # edge_type.append(1)
edge_index = torch.tensor(edges_list, dtype=torch.long).t() edge_index = torch.tensor(edges_list, dtype=torch.long).t()
edge_type = torch.tensor(edge_type, dtype=torch.long) edge_type = torch.tensor(edge_type, dtype=torch.long)