debugging

EP/embedding_propagation.py

@@ -59,14 +59,13 @@ class EmbeddingPropagation:
 						#neighbors = [list(self.graph.neighbors(node)) for node in batch]
 						neighbors = [[x for x in self.graph[node]] for node in batch]
 						print(neighbors)
-						neighbor_labels = np.array([[self.graph.nodes[node][lt] for node in neighs] for neighs in neighbors])
+						neighbor_labels = [np.concatenate([self.graph.nodes[node][lt] for node in neighs], axis=None) for neighs in neighbors]
 
-
-						print(type(neighbor_labels))
+						print(neighbor_labels)
 
 						print('-'*100)
-						#neighbor_labels_lengths = [len(labels_list) for labels_list in neighbor_labels]
-						neighbor_labels_lengths = [[len(labels_list) for labels_list in x] for x in neighbor_labels]
+						neighbor_labels_lengths = [len(labels_list) for labels_list in neighbor_labels]
+						#neighbor_labels_lengths = [[len(labels_list) for labels_list in x] for x in neighbor_labels]
 						print(neighbor_labels_lengths)
 						neighbor_labels_padded = np.array([np.array(pd.DataFrame(label).fillna(0).values.astype(np.int32)) for label in neighbor_labels])
 						print(neighbor_labels_padded.shape)