bug fix

safe-graph · Feb 26, 2022 · 3a0102e · 3a0102e
1 parent 4d017a5
commit 3a0102e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/algorithms/HACUD/get_data.py b/algorithms/HACUD/get_data.py
@@ -32,8 +32,8 @@ def __init__(self, path, save_path):
         for i in range(self.n_metapath):
             z = self.rownetworks[i]
             adj.append(z)
-            u_index.append(z[0])
-            v_index.append(z[1])   
+            u_index.append(np.where(z)[0])
+            v_index.append(np.where(z)[1])   
             self.n_int.append(len(np.where(z)[0]))
 
         self.print_statistics()
@@ -210,4 +210,4 @@ def create_sparsity_split(self):
 
 # if __name__ == '__main__':
 #     path = "../../dataset/DBLP4057_GAT_with_idx_tra200_val_800.mat"
-#     data_generator = Data(path=path, save_path = save_path)
+#     data_generator = Data(path=path, save_path = save_path)