fix error in struc2vec

core/Embedding/ge/models/struc2vec.py

@@ -35,6 +35,7 @@
 from ..walker import BiasedWalker
 from pathlib import Path
 
+
 class Struc2Vec():
     def __init__(self, 
                  graph, 
@@ -152,7 +153,7 @@ def _compute_ordered_degreelist(self, max_num_layers):
 
         degreeList = {}
         vertices = self.idx  # self.g.nodes()
-        for v in vertices:
+        for v in tqdm(vertices):
             degreeList[v] = self._get_order_degreelist_node(v, max_num_layers)
         return degreeList
 
@@ -224,7 +225,7 @@ def _compute_structural_distance(self, max_num_layers, workers=1, verbose=0,):
                 degreeListsSelected = {}
                 vertices = {}
                 n_nodes = len(self.idx)
-                for v in self.idx:  # c:list of vertex
+                for v in tqdm(self.idx):  # c:list of vertex
                     nbs = get_vertices(
                         v, len(self.graph[self.idx2node[v]]), degrees, n_nodes)
                     vertices[v] = nbs  # store nbs