changes tp helper scripts

xiwang0818 · Apr 13, 2022 · 7b963e1 · 7b963e1
1 parent d409acd
commit 7b963e1
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 10 deletions.
diff --git a/__pycache__/helper.cpython-39.pyc b/__pycache__/helper.cpython-39.pyc
diff --git a/helper.py b/helper.py
@@ -63,13 +63,16 @@ def train_test_loader(data_list, graph_data):
 def get_data(graph_data, folder_path):
     data_list = []
     for file_ in graph_data:
-        G=nx.read_gpickle(f"{folder_path}/{file_}")
-        data = get_data_from_graph(G)
-        A = get_adjacency_matrix(G)
-        data.x = torch.from_numpy(A).float()
-        data.y = return_labels(G)
-        data.train_mask, data.val_mask, data.test_mask = retrieve_masks(data.y)
-        data_list.append(data)
+        if 'gpickle' in file_:
+            print(file_)
+            G=nx.read_gpickle(f"{folder_path}/{file_}")
+            print(G)
+            data = get_data_from_graph(G)
+            A = get_adjacency_matrix(G)
+            data.x = torch.from_numpy(A).float()
+            data.y = return_labels(G)
+            data.train_mask, data.val_mask, data.test_mask = retrieve_masks(data.y)
+            data_list.append(data)
     return data_list
 
 def report_training_accuracy(accuracy_dict):

diff --git a/logger.txt b/logger.txt
@@ -1 +1 @@
-evaluate.py -> Imports are successful.
+train.py -> Imports are successful.
diff --git a/test_data/readme.txt b/test_data/readme.txt
@@ -1 +0,0 @@
-Storing the test data.

diff --git a/train_data/readme.txt b/train_data/readme.txt
@@ -1 +0,0 @@
-Storing the training data.