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

process function needs 2 dimension for y

Browse Source
This commit is contained in:
mhz
2024-07-01 10:06:51 +02:00
parent 4d1dea1179
commit 0c3cfb195a
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

@@ -674,7 +674,7 @@ class Dataset(InMemoryDataset):
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)
edge_attr = edge_type edge_attr = edge_type
y = torch.tensor([0], dtype=torch.float).view(1, -1) y = torch.tensor([0, 0], dtype=torch.float).view(1, -1)
data = Data(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y, idx=i) data = Data(x=x, edge_index=edge_index, edge_attr=edge_attr, y=y, idx=i)
return data return data
graph_list = [] graph_list = []
@@ -898,7 +898,7 @@ class Dataset_origin(InMemoryDataset):
torch.save(self.collate(data_list), self.processed_paths[0]) torch.save(self.collate(data_list), self.processed_paths[0])
def parse_architecture_string(arch_str): def parse_architecture_string(arch_str):
print(arch_str) # print(arch_str)
steps = arch_str.split('+') steps = arch_str.split('+')
nodes = ['input'] # Start with input node nodes = ['input'] # Start with input node
edges = [] edges = []