(graph)
| 1 | def preprocess_nxgraph(graph): |
| 2 | node2idx = {} |
| 3 | idx2node = [] |
| 4 | node_size = 0 |
| 5 | for node in graph.nodes(): |
| 6 | node2idx[node] = node_size |
| 7 | idx2node.append(node) |
| 8 | node_size += 1 |
| 9 | return idx2node, node2idx |
| 10 | |
| 11 | |
| 12 | def partition_dict(vertices, workers): |